1 files changed, 5 insertions, 5 deletions
diff --git a/src/test/java/com/seaofnodes/simple/Chapter17aTest.java b/src/test/java/com/seaofnodes/simple/Chapter17aTest.java
index ba5bb985..d03248c2 100644
--- a/src/test/java/com/seaofnodes/simple/Chapter17aTest.java
+++ b/src/test/java/com/seaofnodes/simple/Chapter17aTest.java
@@ -15,7 +15,7 @@ int x=2, y=3;
 if( arg ) { int x = y; x = x*x; y=x; } // Shadow final x
 return y;
 """);
-        code.parse().iter().finalizeIR();
+        code.driver(CodeGen.Phase.FinalizeIR);
         assertEquals("return Phi(Region,9,3);", code.print());
         assertEquals("3", Eval2.eval(code, 0));
         assertEquals("9", Eval2.eval(code, 1));
@@ -28,7 +28,7 @@ struct X { int ~x; };
 X z = new X { x=3; };
 return z.x;
 """);
-        code.parse().iter().finalizeIR();
+        code.driver(CodeGen.Phase.FinalizeIR);
         assertEquals("return 3;", code.print());
         assertEquals("3", Eval2.eval(code,  0));
     }
@@ -40,7 +40,7 @@ struct Point { int ~x, ~y; };
 Point p = new Point { x=3; y=4; };
 return p;
 """);
-        code.parse().iter().finalizeIR();
+        code.driver(CodeGen.Phase.FinalizeIR);
         assertEquals("return (const)Point;", code.print());
         assertEquals("Point{x=3,y=4}", Eval2.eval(code,  0));
     }
@@ -73,7 +73,7 @@ return p;
     public void testStructFinal3() {
         for( String fields : new String[] { "var x; var y;", "int[] !x;", "int[] !x;" } ) {
             String src = "struct Point { "+fields+" }; Point p = new Point; p.x++; return p;";
-            try { new CodeGen(src).parse().iter().finalizeIR(); fail(fields); }
+            try { new CodeGen(src).driver(CodeGen.Phase.FinalizeIR); fail(fields); }
             catch( Exception e ) { assertEquals("'Point' is not fully initialized, field 'x' needs to be set in a constructor",e.getMessage()); }
         }
     }
@@ -98,7 +98,7 @@ struct Point { var x=3; var y=4; };
 p.x++;
 return p;
 """);
-        code.parse().iter().finalizeIR();
+        code.driver(CodeGen.Phase.FinalizeIR);
         assertEquals("return Point;", code.print());
         assertEquals("Point{x=4,y=4}", Eval2.eval(code,  0));
     }