diff --git a/src/babel/generation/index.js b/src/babel/generation/index.js
index 90518b17e6..4870efe5b0 100644
--- a/src/babel/generation/index.js
+++ b/src/babel/generation/index.js
@@ -21,7 +21,7 @@ class CodeGenerator {
     this.opts     = opts;
     this.ast      = ast;
 
-    this.whitespace = new Whitespace(this.tokens, this.comments, this.format);
+    this.whitespace = new Whitespace(this.tokens);
     this.position   = new Position;
     this.map        = new SourceMap(this.position, opts, code);
     this.buffer     = new Buffer(this.position, this.format);
diff --git a/src/babel/generation/whitespace.js b/src/babel/generation/whitespace.js
index f4cf563b47..90709597e0 100644
--- a/src/babel/generation/whitespace.js
+++ b/src/babel/generation/whitespace.js
@@ -1,5 +1,3 @@
-import sortBy from "lodash/collection/sortBy";
-
 /**
  * Returns `i`th number from `base`, continuing from 0 when `max` is reached.
  * Useful for shifting `for` loop by a fixed number but going over all items.
@@ -21,7 +19,7 @@ function getLookupIndex(i, base, max) {
 }
 
 export default class Whitespace {
-  constructor(tokens, comments) {
+  constructor(tokens) {
     this.tokens = tokens;
     this.used   = {};