diff --git a/angular.json b/angular.json
index af11b3e..be8c9ee 100644
--- a/angular.json
+++ b/angular.json
@@ -105,8 +105,11 @@
           }
         },
         "test": {
-          "builder": "@angular-devkit/build-angular:karma",
+          "builder": "@angular-builders/custom-webpack:karma",
           "options": {
+            "customWebpackConfig": {
+              "path": "./webpack.config.js"
+            },
             "main": "src/test.ts",
             "polyfills": "src/polyfills.ts",
             "tsConfig": "src/tsconfig.spec.json",
diff --git a/webpack.config.js b/webpack.config.js
index 8eeae31..d57a46e 100644
--- a/webpack.config.js
+++ b/webpack.config.js
@@ -1,4 +1,5 @@
 const WebpackNotifierPlugin = require('webpack-notifier');
+const webpack = require('webpack');
 const path = require('path');
 
 module.exports = {
@@ -19,7 +20,7 @@ module.exports = {
         exclude: /node_modules\/(?!(dmn-js|dmn-js-drd|dmn-js-shared|dmn-js-decision-table|table-js|dmn-js-literal-expression|diagram-js)\/).*/,
         loader: 'babel-loader',
         options: {
-          presets: ["@babel/preset-react", '@babel/preset-env']
+          presets: ['@babel/preset-react', '@babel/preset-env']
         }
       }
     ]
@@ -29,6 +30,6 @@ module.exports = {
       alwaysNotify: true,
       title: 'App Name',
       contentImage: path.join(__dirname, 'image.png')
-    }),
+    })
   ]
 };