diff --git a/src/build.rs b/src/build.rs
index f783584..6c476ad 100644
--- a/src/build.rs
+++ b/src/build.rs
@@ -1201,7 +1201,7 @@ mod compile_test {
 #[cfg(test)]
 mod test {
     use super::{Builder, CallDef, MacroDef, SelectDef, Val};
-    use ast::*;
+    use ast::tree::*;
     use std::rc::Rc;
 
     fn test_expr_to_val(mut cases: Vec<(Expression, Val)>, b: Builder) {
diff --git a/src/format.rs b/src/format.rs
index 44ec36e..bff58c7 100644
--- a/src/format.rs
+++ b/src/format.rs
@@ -77,7 +77,7 @@ impl<V: Into<String> + Clone> Formatter<V> {
 #[cfg(test)]
 mod test {
     use super::Formatter;
-    use ast::Position;
+    use ast::tree::Position;
 
     #[test]
     fn test_format_happy_path() {