diff --git a/ast-generator/src/build_fn/enum_build_fn.rs b/ast-generator/src/build_fn/tree_enum_build_fn.rs
similarity index 88%
rename from ast-generator/src/build_fn/enum_build_fn.rs
rename to ast-generator/src/build_fn/tree_enum_build_fn.rs
index 296f6e4..2dfe577 100644
--- a/ast-generator/src/build_fn/enum_build_fn.rs
+++ b/ast-generator/src/build_fn/tree_enum_build_fn.rs
@@ -1,9 +1,9 @@
-use crate::spec::{EnumBuildSpec, EnumRuleChildKind};
-use crate::util::{make_build_fn_name, make_build_pair};
+use crate::deserialize::util::{make_build_fn_name, make_build_pair};
+use crate::spec::tree_enum_spec::{EnumRuleChildKind, TreeEnumBuildSpec};
 use proc_macro2::TokenStream;
 use quote::{format_ident, quote};
 
-pub fn make_enum_build_fn(enum_build_spec: &EnumBuildSpec) -> TokenStream {
+pub fn make_enum_build_fn(enum_build_spec: &TreeEnumBuildSpec) -> TokenStream {
     let build_fn_ident = format_ident!("{}", make_build_fn_name(enum_build_spec.build()));
     let pair_ident = format_ident!("{}", make_build_pair(enum_build_spec.build()));
     let return_type_ident = format_ident!("{}", enum_build_spec.build());