1 files changed, 5 insertions, 5 deletions
diff --git a/lib/generate/index.js b/lib/generate/index.js
index c8b98ef..5f44633 100644
--- a/lib/generate/index.js
+++ b/lib/generate/index.js
@@ -127,13 +127,13 @@ var generate = function(options) {
                     .value()
                 );
             })
+
+            // Finish gneration
+            .then(function() {
+                return generator.finish();
+            });
         }
     })
-
-    // Finish gneration
-    .then(function() {
-        return generator.finish();
-    });
 };
 
 module.exports = {