diff --git a/test/index.js b/test/index.js
index 55160c7..fafa9d8 100644
--- a/test/index.js
+++ b/test/index.js
@@ -41,7 +41,8 @@ test.cb('EVERYTHING WORKS', (t) => {
       { test: /\.txt$/, loader: 'source' }
     ]},
     ignore: ['**/views/ignoreme.txt'],
-    plugins: [plugin]
+    plugins: [plugin],
+    devtool: 'source-map'
   })
 
   project.on('error', t.fail)
diff --git a/test/plugin.js b/test/plugin.js
index 4dff87f..e50c94b 100644
--- a/test/plugin.js
+++ b/test/plugin.js
@@ -24,7 +24,7 @@ module.exports = class TestPlugin extends EventEmitter {
 
     compiler.plugin('compilation', (compilation) => {
       compilation.plugin('optimize-chunk-assets', (chunks, done) => {
-        this.util.removeAssets(compilation, this.injectFile)
+        this.util.removeAssets(compilation, this.injectFile, chunks)
         this.emit('removeAssets')
         done()
       })