diff --git a/src/test/java/de/dhbw/horb/StatementGeneratorTest.java b/src/test/java/de/dhbw/horb/StatementGeneratorTest.java
index 4229bf1..7fcca05 100644
--- a/src/test/java/de/dhbw/horb/StatementGeneratorTest.java
+++ b/src/test/java/de/dhbw/horb/StatementGeneratorTest.java
@@ -27,6 +27,6 @@ public class StatementGeneratorTest
         DecafLexer lexer = new DecafLexer(input);
         CommonTokenStream tokens = new CommonTokenStream(lexer);
         DecafParser parser = new DecafParser(tokens);
-        return new StatementGenerator().visit(parser.stmt());
+        return parser.stmt().accept(new StatementGenerator());
     }
 }
\ No newline at end of file