1 files changed, 4 insertions, 3 deletions
diff --git a/lib/template/index.js b/lib/template/index.js
index b59dd04..3f74267 100644
--- a/lib/template/index.js
+++ b/lib/template/index.js
@@ -21,9 +21,10 @@ function normBlockResult(blk) {
     return blk;
 }
 
-function TemplateEngine(book) {
-    this.book = book;
-    this.log = book.log;
+function TemplateEngine(output) {
+    this.output = output;
+    this.book = output.book;
+    this.log = this.book.log;
 
     // Create file loader
     this.loader = new Loader(this);