diff --git a/test/generation.js b/test/generation.js
index 4a63b26023..d4ee3a36c9 100644
--- a/test/generation.js
+++ b/test/generation.js
@@ -27,7 +27,7 @@ _.each(helper.get("generation"), function (testSuite) {
         var actual = task.actual;
 
         var actualAst  = util.parse({ filename: actual.loc }, actual.code);
-        var actualCode = generate(actual.code, actualAst).code;
+        var actualCode = generate(actualAst, null, actual.code).code;
 
         chai.expect(actualCode).to.equal(expect.code, actual.loc + " !== " + expect.loc);
       });