diff --git a/ql/src/ql.qll b/ql/src/ql.qll
index 89e435bca97..496c4d1f595 100644
--- a/ql/src/ql.qll
+++ b/ql/src/ql.qll
@@ -1,2 +1,3 @@
+import codeql.Locations
 import codeql.files.FileSystem
-import codeql_ql.ast.internal.TreeSitter
+import codeql_ql.ast.internal.TreeSitter::Generated