diff --git a/openmetadata-ui-core-components/src/main/resources/ui/vite.config.ts b/openmetadata-ui-core-components/src/main/resources/ui/vite.config.ts
index 0a6c794bef6..40c9dd82564 100644
--- a/openmetadata-ui-core-components/src/main/resources/ui/vite.config.ts
+++ b/openmetadata-ui-core-components/src/main/resources/ui/vite.config.ts
@@ -26,6 +26,8 @@ export default defineConfig({
         'react/jsx-runtime',
         '@mui/material',
         '@mui/system',
+        '@mui/material/styles',
+        '@mui/material/Chip',
         '@mui/icons-material',
         '@mui/x-date-pickers',
         '@emotion/react',