1 files changed, 6 insertions, 3 deletions
diff --git a/spec/env/node.js b/spec/env/node.js
index fe34f94..3517479 100644
--- a/spec/env/node.js
+++ b/spec/env/node.js
@@ -1,13 +1,16 @@
 require('./common');
 
-global.Handlebars = require('../../lib/handlebars');
+global.Handlebars = require('../../zomg/lib/handlebars');
 
 global.CompilerContext = {
-  compile: function(template, options) {
+  compile: function(template, options, env) {
+    env = env || handlebarsEnv;
     var templateSpec = Handlebars.precompile(template, options);
-    return Handlebars.template(eval('(' + templateSpec + ')'));
+    return Handlebars.template(eval('(' + templateSpec + ')'), env);
   },
   compileWithPartial: function(template, options) {
+    options = options || {};
+    options.env = handlebarsEnv;
     return Handlebars.compile(template, options);
   }
 };