1 files changed, 3 insertions, 3 deletions
diff --git a/src/test/java/com/seaofnodes/simple/SchedulerTest.java b/src/test/java/com/seaofnodes/simple/SchedulerTest.java
index ca211c60..6156700d 100644
--- a/src/test/java/com/seaofnodes/simple/SchedulerTest.java
+++ b/src/test/java/com/seaofnodes/simple/SchedulerTest.java
@@ -51,7 +51,7 @@ int i=v.f;
 if (arg) v.f=1;
 return i;
 """);
-        code.parse();//.iter().finalizeIR();
+        code.parse();//.iter().opto().iterPostOpto().finalizeIR();
         var eval = new Evaluator(code._stop);
         assertEquals(2L, eval.evaluate(1, 10));
     }
@@ -66,7 +66,7 @@ struct S {
 if(arg) v0.f=1;
 return v0;
 """);
-        code.parse().iter().finalizeIR();
+        code.driver(CodeGen.Phase.FinalizeIR);
         var eval = new Evaluator(code._stop);
         assertObj(eval.evaluate(0, 10), "S", 0L);
         assertObj(eval.evaluate(1, 10), "S", 1L);
@@ -85,7 +85,7 @@ i=v.f;
 if (arg) v.f=1;
 return i;
 """);
-        code.parse().iter().finalizeIR();
+        code.driver(CodeGen.Phase.FinalizeIR);
         var eval = new Evaluator(code._stop);
         assertEquals(2L, eval.evaluate(0, 10));
         assertEquals(2L, eval.evaluate(1, 10));