1 files changed, 4 insertions, 3 deletions
diff --git a/build.rs b/build.rs
index 624be282..b9cf3e60 100644
--- a/build.rs
+++ b/build.rs
@@ -5,11 +5,12 @@ mod codegen {
 
     pub fn main() {
         let out_dir = env::var_os("OUT_DIR").unwrap();
-        let src = Path::new("src/gen.rs");
-        let dst = Path::new(&out_dir).join("gen.rs");
+        let src = Path::new("src/codegen/mod.rs");
+        let dst = Path::new(&out_dir).join("codegen.rs");
 
         quasi_codegen::expand(&src, &dst).unwrap();
-        println!("cargo:rerun-if-changed=src/gen.rs");
+        println!("cargo:rerun-if-changed=src/codegen/mod.rs");
+        println!("cargo:rerun-if-changed=src/codegen/helpers.rs");
     }
 }