diff --git a/packages/builder/vite.config.js b/packages/builder/vite.config.js
index 0858f4c496..18ce02fe8f 100644
--- a/packages/builder/vite.config.js
+++ b/packages/builder/vite.config.js
@@ -134,6 +134,10 @@ export default defineConfig(({ mode }) => {
           find: "@budibase/shared-core",
           replacement: path.resolve("../shared-core/src"),
         },
+        {
+          find: "@budibase/bbui",
+          replacement: path.resolve("../bbui/src"),
+        },
       ],
     },
   }