diff --git a/lib/resources/tasks/build-webpack.js b/lib/resources/tasks/build-webpack.js
index 7028d32f7..728e4f52a 100644
--- a/lib/resources/tasks/build-webpack.js
+++ b/lib/resources/tasks/build-webpack.js
@@ -23,7 +23,7 @@ function buildWebpack(done) {
     compiler.watch({}, onBuild);
   } else {
     compiler.run(onBuild);
-    compiler.plugin('done', () => done());
+    compiler.hooks.done.tap('AureliaCLI', done);
   }
 }
 
diff --git a/lib/resources/tasks/build-webpack.ts b/lib/resources/tasks/build-webpack.ts
index 38d720e0d..440a7f26f 100644
--- a/lib/resources/tasks/build-webpack.ts
+++ b/lib/resources/tasks/build-webpack.ts
@@ -23,7 +23,7 @@ function buildWebpack(done) {
     compiler.watch({}, onBuild);
   } else {
     compiler.run(onBuild);
-    compiler.plugin('done', () => done());
+    compiler.hooks.done.tap('AureliaCLI', done);
   }
 }