1 files changed, 9 insertions, 9 deletions
diff --git a/src/test/java/com/seaofnodes/simple/Chapter07Test.java b/src/test/java/com/seaofnodes/simple/Chapter07Test.java
index daca77ab..7094c0d5 100644
--- a/src/test/java/com/seaofnodes/simple/Chapter07Test.java
+++ b/src/test/java/com/seaofnodes/simple/Chapter07Test.java
@@ -17,8 +17,8 @@ while(arg < 10) {
 }
 return arg;
 """);
-        code.parse();
-        assertEquals("return Phi(Loop,arg,(Convert_i64(Phi_arg)+1));", code.print());
+        code.driver(CodeGen.Phase.FinalizeIR);
+        assertEquals("return Phi(Loop,arg,(Phi_arg+1));", code.print());
         assertTrue(code.ctrl() instanceof CProjNode);
     }
 
@@ -71,7 +71,7 @@ while(a < 10) {
 return b;
 """);
         code.parse();
-        assertEquals("return Convert_i64(Phi(Loop,2,Phi(Region,Phi_b,4)));", code.print());
+        assertEquals("return Phi(Loop,2,Phi(Region,Phi_b,4));", code.print());
         assertTrue(code.ctrl() instanceof CProjNode);
     }
 
@@ -106,8 +106,8 @@ while(a < 10) {
 }
 return a;
 """);
-        code.parse();
-        assertEquals("return Phi(Loop,1,(Convert_i64(Phi_a)+3));", code.print());
+        code.driver(CodeGen.Phase.FinalizeIR);
+        assertEquals("return Phi(Loop,1,(Phi_a+3));", code.print());
         assertTrue(code.ctrl() instanceof CProjNode);
     }
 
@@ -164,8 +164,8 @@ while(a < 10) {
 }
 return a;
 """);
-        code.parse();
-        assertEquals("return Phi(Loop,1,(Convert_i64(Phi_a)+3));", code.print());
+        code.driver(CodeGen.Phase.FinalizeIR);
+        assertEquals("return Phi(Loop,1,(Phi_a+3));", code.print());
         assertTrue(code.ctrl() instanceof CProjNode);
     }
 
@@ -197,8 +197,8 @@ while(a < 10) {
 }
 return a;
 """);
-        code.parse();
-        assertEquals("return Phi(Loop,1,(Convert_i64(Phi_a)+3));", code.print());
+        code.driver(CodeGen.Phase.FinalizeIR);
+        assertEquals("return Phi(Loop,1,(Phi_a+3));", code.print());
         assertTrue(code.ctrl() instanceof CProjNode);
     }