diff --git a/codeql-custom-queries-ruby/example.ql b/codeql-custom-queries-ruby/example.ql
index d086464..dcfb81e 100644
--- a/codeql-custom-queries-ruby/example.ql
+++ b/codeql-custom-queries-ruby/example.ql
@@ -6,6 +6,7 @@
  */
 
 import ruby
+import codeql.ruby.AST
 
 from Block b
 where b.getNumberOfStatements() = 0