diff --git a/d2ir/import_test.go b/d2ir/import_test.go
index 0b9497abe..76012a33a 100644
--- a/d2ir/import_test.go
+++ b/d2ir/import_test.go
@@ -157,7 +157,7 @@ label: meow`,
 					"a.d2":     "vars: { x: 2 }; hi: ${x}",
 				})
 				assert.Success(t, err)
-				assertQuery(t, m, 0, 0, 2, "hi")
+				assertQuery(t, m, 0, 0, "2", "hi")
 			},
 		},
 		{
@@ -168,7 +168,7 @@ label: meow`,
 					"a.d2":     "vars: { x: 2 }",
 				})
 				assert.Success(t, err)
-				assertQuery(t, m, 0, 0, 1, "hi")
+				assertQuery(t, m, 0, 0, "1", "hi")
 			},
 		},
 	}