diff --git a/webpack.lib.js b/webpack.lib.js
index 1b70333b4..4dbd3ea2d 100644
--- a/webpack.lib.js
+++ b/webpack.lib.js
@@ -65,7 +65,7 @@ module.exports = [
         KedroViz: './src/utils/viz-entry.js', // Entry point for KedroViz
     },
     output: {
-        path: path.resolve(__dirname, 'dist'),
+        path: path.resolve(__dirname, 'lib/umd'),
         filename: '[name].bundle.min.js',
         library: 'KedroVizBundle', // Name of the UMD library
         libraryTarget: 'umd', // UMD allows compatibility across environments