1 files changed, 2 insertions, 5 deletions
diff --git a/chapter01/src/test/java/com/seaofnodes/simple/ParserTest.java b/chapter01/src/test/java/com/seaofnodes/simple/ParserTest.java
index ad420ece..f1a82ec1 100644
--- a/chapter01/src/test/java/com/seaofnodes/simple/ParserTest.java
+++ b/chapter01/src/test/java/com/seaofnodes/simple/ParserTest.java
@@ -1,9 +1,6 @@
 package com.seaofnodes.simple;
 
-import com.seaofnodes.simple.node.ConstantNode;
-import com.seaofnodes.simple.node.Node;
-import com.seaofnodes.simple.node.ReturnNode;
-import com.seaofnodes.simple.node.StartNode;
+import com.seaofnodes.simple.node.*;
 import org.junit.Assert;
 import org.junit.Test;
 
@@ -11,7 +8,7 @@ public class ParserTest {
 
     @Test
     public void testSimpleProgram() {
-        Parser parser = new Parser();
+        Parser parser = new Parser(new NodeIDGenerator());
         StartNode startNode = parser.parse("return 1;");
         Assert.assertNotNull(startNode);
         Assert.assertTrue(startNode.isControl());