2 files changed, 4 insertions, 4 deletions
diff --git a/test/mock.js b/test/mock.js
index ce7b027..127e35c 100644
--- a/test/mock.js
+++ b/test/mock.js
@@ -4,7 +4,6 @@ var tmp = require('tmp');
 var path = require('path');
 
 var Book = require('../').Book;
-var Output = require('../lib/output');
 var NodeFS = require('../lib/fs/node');
 
 require('./assertions');
@@ -58,7 +57,7 @@ function setupDefaultBook(files, opts) {
 }
 
 // Output a book with a specific generator
-function outputDefaultBook(generator, files, opts) {
+function outputDefaultBook(Output, files, opts) {
     return setupDefaultBook(files, opts)
     .then(function(book) {
         // Parse the book
@@ -66,7 +65,7 @@ function outputDefaultBook(generator, files, opts) {
 
         // Start generation
         .then(function() {
-            var output = new Output(book, generator);
+            var output = new Output(book);
             return output.generate()
                 .thenResolve(output);
         });
diff --git a/test/output-json.js b/test/output-json.js
index b2387a7..6eb3c48 100644
--- a/test/output-json.js
+++ b/test/output-json.js
@@ -1,4 +1,5 @@
 var mock = require('./mock');
+var JSONOutput = require('../lib/output/json');
 
 describe('JSON Output', function() {
 
@@ -6,7 +7,7 @@ describe('JSON Output', function() {
         var output;
 
         before(function() {
-            return mock.outputDefaultBook('json')
+            return mock.outputDefaultBook(JSONOutput)
             .then(function(_output) {
                 output = _output;
             });