From e95f61985e78be7d7e3b32b2dc93e4b48391ddbd Mon Sep 17 00:00:00 2001
From: Alexander Wang <alex@terrastruct.com>
Date: Thu, 13 Jul 2023 14:41:06 -0700
Subject: [PATCH] fix import test

---
 d2ir/import_test.go | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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")
 			},
 		},
 	}