diff --git a/src/tests.rs b/src/tests.rs
index 2b4717e..9f51835 100644
--- a/src/tests.rs
+++ b/src/tests.rs
@@ -12,7 +12,7 @@ fn parse_ast_has_error(filename: &str) {
     println!("Testing {:?}", filename);
     let program = std::fs::read_to_string(filename).expect(filename);
     let ast = get_ast(&program);
-    let res = parse_ast(&ast, false);
+    let res = parse_ast(&ast, false, false);
     assert!(res.is_err());
     insta::assert_debug_snapshot!(res.unwrap_err());
 }
@@ -21,7 +21,7 @@ fn parse_ast_is_ok(filename: &str) -> QuadrupleManager {
     println!("Testing {:?}", filename);
     let program = std::fs::read_to_string(filename).expect(filename);
     let ast = get_ast(&program);
-    let res = parse_ast(&ast, false);
+    let res = parse_ast(&ast, false, false);
     assert!(res.is_ok());
     let quad_manager = res.unwrap();
     insta::assert_display_snapshot!(quad_manager);