1 files changed, 3 insertions, 3 deletions
diff --git a/main.ml b/main.ml
index e98dd5f..b63b35f 100644
--- a/main.ml
+++ b/main.ml
@@ -31,7 +31,7 @@ let options = [
 ]
 
 let do_test_interpret prog verbose =
-    let s0 = Interpret.init_state prog "test" in
+    let s0 = Interpret.init_state (Typing.root_prog prog "test") in
     if verbose then begin
       Format.printf "Init state:@.";
       Interpret.print_state Format.std_formatter s0;
@@ -75,8 +75,8 @@ let () =
       let prog = Rename.rename_prog prog in
       if !dumprn then Ast_printer.print_prog Format.std_formatter prog;
 
-      if !ai_itv then AI_Itv.do_prog !ai_widen_delay prog !ai_root;
-      if !ai_rel then AI_Rel.do_prog !ai_widen_delay prog !ai_root;
+      if !ai_itv then AI_Itv.do_prog !ai_widen_delay (Typing.root_prog prog !ai_root);
+      if !ai_rel then AI_Rel.do_prog !ai_widen_delay (Typing.root_prog prog !ai_root);
 
       if !vtest then do_test_interpret prog true
       else if !test then do_test_interpret prog false;