1 files changed, 4 insertions, 2 deletions
diff --git a/lib/output/generateBook.js b/lib/output/generateBook.js
index c5d61f8..4e64eb0 100644
--- a/lib/output/generateBook.js
+++ b/lib/output/generateBook.js
@@ -156,8 +156,7 @@ function generateBook(generator, book, options) {
     var state = generator.State? generator.State({}) : Immutable.Map();
     var start = Date.now();
 
-    return processOutput(
-        generator,
+    return Promise(
         new Output({
             book: book,
             options: options,
@@ -165,6 +164,9 @@ function generateBook(generator, book, options) {
             generator: generator.name
         })
     )
+    .then(processOutput.bind(null, generator))
+
+    // Log duration and end message
     .then(function(output) {
         var logger = output.getLogger();
         var end = Date.now();