2 files changed, 5 insertions, 23 deletions
diff --git a/testing/Cargo.toml b/testing/Cargo.toml
index 6acdaf5..449cc6b 100644
--- a/testing/Cargo.toml
+++ b/testing/Cargo.toml
@@ -8,3 +8,6 @@ build = "build.rs"
 [dependencies]
 askama = { path = "../askama" }
 askama_derive = { path = "../askama_derive" }
+
+[build-dependencies]
+askama = { path = "../askama" }
diff --git a/testing/build.rs b/testing/build.rs
index 2f868f4..89e3e6b 100644
--- a/testing/build.rs
+++ b/testing/build.rs
@@ -1,26 +1,5 @@
-use std::env;
-use std::fs::{self, DirEntry};
-use std::io;
-use std::path::Path;
-
-fn visit_dirs(dir: &Path, cb: &Fn(&DirEntry)) -> io::Result<()> {
-    if dir.is_dir() {
-        for entry in try!(fs::read_dir(dir)) {
-            let entry = try!(entry);
-            let path = entry.path();
-            if path.is_dir() {
-                try!(visit_dirs(&path, cb));
-            } else {
-                cb(&entry);
-            }
-        }
-    }
-    Ok(())
-}
+extern crate askama;
 
 fn main() {
-    let root = env::var("CARGO_MANIFEST_DIR").unwrap();
-    visit_dirs(&Path::new(&root).join("templates"), &|e: &DirEntry| {
-        println!("cargo:rerun-if-changed={}", e.path().to_str().unwrap());
-    }).unwrap();
+    askama::rerun_if_templates_changed();
 }