1 files changed, 8 insertions, 1 deletions
diff --git a/lib/output/generateBook.js b/lib/output/generateBook.js
index ab07e41..ef7d5f1 100644
--- a/lib/output/generateBook.js
+++ b/lib/output/generateBook.js
@@ -1,3 +1,5 @@
+var Immutable = require('immutable');
+
 var Output = require('../models/output');
 var Config = require('../models/config');
 var Promise = require('../utils/promise');
@@ -33,10 +35,15 @@ var generatePages = require('./generatePages');
 */
 function generateBook(generator, book, options) {
     options = generator.Options(options);
+    var state = generator.State? generator.State({}) : Immutable.Map();
     var start = Date.now();
 
     return Promise(
-        Output.createForBook(book, options)
+        new Output({
+            book: book,
+            options: options,
+            state: state
+        })
     )
     .then(preparePlugins)
     .then(preparePages)