diff --git a/ast-generator/src/lib.rs b/ast-generator/src/lib.rs
index 85c534c..60a496b 100644
--- a/ast-generator/src/lib.rs
+++ b/ast-generator/src/lib.rs
@@ -127,6 +127,13 @@ fn generate_build_file(build_specs: &[BuildSpec]) -> AstGeneratedFile {
         })
         .collect::<Vec<_>>();
     let combined = quote! {
+        //noinspection RsUnusedImport
+        use crate::parser::Rule;
+        //noinspection RsUnusedImport
+        use pest::iterators::Pair;
+        //noinspection RsUnusedImport
+        use crate::ast::node::*;
+        
         #(#build_fns)*
     };
     AstGeneratedFile {
diff --git a/src/ast/mod.rs b/src/ast/mod.rs
index 443511d..6508bea 100644
--- a/src/ast/mod.rs
+++ b/src/ast/mod.rs
@@ -45,13 +45,6 @@ pub mod node {
 }
 
 pub mod build {
-    //noinspection RsUnusedImport
-    use crate::parser::Rule;
-    //noinspection RsUnusedImport
-    use pest::iterators::Pair;
-    //noinspection RsUnusedImport
-    use crate::ast::node::*;
-
     include!(concat!(env!("OUT_DIR"), "/src/ast/build.rs"));
 
     #[cfg(test)]