improved testing.

1 files changed, 7 insertions, 8 deletions

diff --git a/lib/logger.js b/lib/logger.js
index f44fe28..985b67b 100644
--- a/lib/logger.js
+++ b/lib/logger.js
@@ -1,9 +1,8 @@
-var logger = require('log-driver')({level : 'warn'});
-
-if (process.argv[2]) {
-  if (~['-v', '--verbose'].indexOf(process.argv[2])) {
-    logger = require('log-driver')({level : 'debug'});
+module.exports = function(){
+  if (process.argv[2]) {
+    if (~['-v', '--verbose'].indexOf(process.argv[2])) {
+      return require('log-driver')({level : 'debug'});
+    }
   }
-}
-
-module.exports = logger;
+  return require('log-driver')({level : 'warn'});
+};