Enable extension "do"

1 files changed, 7 insertions, 2 deletions

diff --git a/lib/output/website/createTemplateEngine.js b/lib/output/website/createTemplateEngine.js
index b90c3ad..5c5acee 100644
--- a/lib/output/website/createTemplateEngine.js
+++ b/lib/output/website/createTemplateEngine.js
@@ -1,4 +1,6 @@
 var path = require('path');
+var nunjucks = require('nunjucks');
+var DoExtension = require('nunjucks-do')(nunjucks);
 
 var TEMPLATES_FOLDER = require('../../constants/templatesFolder');
 
@@ -27,8 +29,11 @@ function createTemplateEngine(output) {
 
     var loader = new Templating.ThemesLoader(tplSearchPaths);
 
-    return new TemplateEngine({
-        loader: loader
+    return TemplateEngine.create({
+        loader: loader,
+        extensions: {
+            'DoExtension': new DoExtension()
+        }
     });
 }