diff --git a/tests/mocha/generator_test.js b/tests/mocha/generator_test.js
index 945786c75..8cc5d7901 100644
--- a/tests/mocha/generator_test.js
+++ b/tests/mocha/generator_test.js
@@ -94,6 +94,7 @@ suite('Generator', function() {
         var generator = testCase[0];
         var name = testCase[1];
         test(name, function() {
+          generator.init(this.workspace);
           this.blockToCodeTest(generator, false, true, 'row_block');
           this.blockToCodeTest(
               generator, false, false, 'row_blockstack_block', 'thisOnly=false');