Correctly set template.self in all cases

1 files changed, 4 insertions, 0 deletions

diff --git a/lib/output/website/templateEnv.js b/lib/output/website/templateEnv.js
index ea2b521..b3009db 100644
--- a/lib/output/website/templateEnv.js
+++ b/lib/output/website/templateEnv.js
@@ -2,6 +2,8 @@ var _ = require('lodash');
 var nunjucks = require('nunjucks');
 var path = require('path');
 var fs = require('fs');
+var DoExtension = require('nunjucks-do')(nunjucks);
+
 
 var location = require('../../utils/location');
 var defaultFilters = require('../../template/filters');
@@ -24,6 +26,8 @@ function setupTemplateEnv(output, context) {
     );
     var env = new nunjucks.Environment(loader);
 
+    env.addExtension('DoExtension', new DoExtension());
+
     // Add context as global
     _.each(context, function(value, key) {
         env.addGlobal(key, value);