2 files changed, 4 insertions, 4 deletions
diff --git a/bin/build.js b/bin/build.js
index 49978b8..bea8ff9 100644
--- a/bin/build.js
+++ b/bin/build.js
@@ -47,8 +47,8 @@ var buildFunc = function(dir, options) {
     })
     .then(function(output) {
         console.log("Successfuly built !");
-    }, utils.logError)
-    .then(_.constant(outputDir));
+        return output;
+    }, utils.logError);
 };
 
 module.exports = buildFunc;
 \ No newline at end of file
diff --git a/bin/gitbook.js b/bin/gitbook.js
index ef842fb..136c2a5 100755
--- a/bin/gitbook.js
+++ b/bin/gitbook.js
@@ -42,10 +42,10 @@ buildCommand(prog
 .option('-p, --port <port>', 'Port for server to listen on', 4000),
 function(dir, options) {
     buildFunc(dir, options)
-    .then(function(outputDir) {
+    .then(function(_options) {
         console.log();
         console.log('Starting server ...');
-        return utils.serveDir(outputDir, options.port)
+        return utils.serveDir(_options.output, options.port)
         .fail(utils.logError);
     })
     .then(function() {