diff --git a/ast-generator/src/build_fn/leaf_enum_build_fn.rs b/ast-generator/src/build_fn/leaf_enum_build_fn.rs
index d27ae1c..7844a54 100644
--- a/ast-generator/src/build_fn/leaf_enum_build_fn.rs
+++ b/ast-generator/src/build_fn/leaf_enum_build_fn.rs
@@ -1,5 +1,5 @@
-use crate::spec::LeafEnumBuildSpec;
-use crate::util::{make_build_fn_name, make_build_pair};
+use crate::deserialize::util::{make_build_fn_name, make_build_pair};
+use crate::spec::leaf_enum_spec::LeafEnumBuildSpec;
 use proc_macro2::TokenStream;
 use quote::{format_ident, quote};
 
@@ -11,7 +11,7 @@ pub fn make_leaf_enum_build_fn(leaf_enum_build_spec: &LeafEnumBuildSpec) -> Toke
     let rule_branches = leaf_enum_build_spec
         .rules()
         .map(|leaf_enum_rule| {
-            let rule_ident = format_ident!("{}", leaf_enum_rule.rule());
+            let rule_ident = format_ident!("{}", leaf_enum_rule);
             quote! {
                 Rule::#rule_ident => #return_type_ident::#rule_ident
             }