diff --git a/loki/schemas/queries/loki-log-query/migrate/migrate.cue b/loki/schemas/queries/loki-log-query/migrate/migrate.cue
index be5bb4240..df8087916 100644
--- a/loki/schemas/queries/loki-log-query/migrate/migrate.cue
+++ b/loki/schemas/queries/loki-log-query/migrate/migrate.cue
@@ -15,20 +15,23 @@ package migrate
 
 #target: {
 	datasource: {
-		type: "loki"
+		type?: "loki"
+		uid: string
 		...
 	}
 	expr: string
 	...
 }
 
-kind: "LokiLogQuery"
-spec: {
-	if #target.datasource.uid != _|_ {
-		datasource: {
-			kind: "LokiDatasource"
-			name: #target.datasource.uid
+if #target.datasource.type != _|_ if #target.datasource.type == "loki" {
+	kind: "LokiLogQuery"
+	spec: {
+		if #target.datasource.uid != _|_ {
+			datasource: {
+				kind: "LokiDatasource"
+				name: #target.datasource.uid
+			}
 		}
+		query: #target.expr
 	}
-	query: #target.expr
 }