diff --git a/unified/ql/lib/codeql/unified/Comments.qll b/unified/ql/lib/codeql/unified/Comments.qll
deleted file mode 100644
index e839af2dbee..00000000000
--- a/unified/ql/lib/codeql/unified/Comments.qll
+++ /dev/null
@@ -1,18 +0,0 @@
-/** Provides classes for working with comments. */
-
-private import unified
-
-/**
- * A comment appearing in the source code.
- */
-class Comment extends TriviaToken {
-  // At the moment, comments are the only type trivia token we extract
-  /**
-   * Gets the text inside this comment, not counting the delimeters.
-   */
-  string getCommentText() {
-    result = this.getValue().regexpCapture("//(.*)", 1)
-    or
-    result = this.getValue().regexpCapture("(?s)/\\*(.*)\\*/", 1)
-  }
-}
diff --git a/unified/ql/lib/codeql/unified/internal/AstExtra.qll b/unified/ql/lib/codeql/unified/internal/AstExtra.qll
new file mode 100644
index 00000000000..df98711c975
--- /dev/null
+++ b/unified/ql/lib/codeql/unified/internal/AstExtra.qll
@@ -0,0 +1,22 @@
+/**
+ * Provides additional AST-like classes outside the generated tree-sitter classes.
+ */
+
+private import unified
+
+module Public {
+  /**
+   * A comment appearing in the source code.
+   */
+  class Comment extends TriviaToken {
+    // At the moment, comments are the only type trivia token we extract
+    /**
+     * Gets the text inside this comment, not counting the delimeters.
+     */
+    string getCommentText() {
+      result = this.getValue().regexpCapture("//(.*)", 1)
+      or
+      result = this.getValue().regexpCapture("(?s)/\\*(.*)\\*/", 1)
+    }
+  }
+}
diff --git a/unified/ql/lib/unified.qll b/unified/ql/lib/unified.qll
index 4f7387ef8f1..73af928aa4e 100644
--- a/unified/ql/lib/unified.qll
+++ b/unified/ql/lib/unified.qll
@@ -5,4 +5,4 @@
 import codeql.Locations
 import codeql.files.FileSystem
 import codeql.unified.Ast::Unified
-import codeql.unified.Comments
+import codeql.unified.internal.AstExtra::Public