diff --git a/components/icon/demo/index.vue b/components/icon/demo/index.vue
index bfcba4480..db0b364f1 100644
--- a/components/icon/demo/index.vue
+++ b/components/icon/demo/index.vue
@@ -8,8 +8,8 @@
   </demo-sort>
 </template>
 
-<script>
-import IconDisplay from '../../../theme/template/IconDisplay';
+<script lang="ts">
+import IconDisplay from '../../../site/src/theme/template/IconDisplay';
 import Basic from './basic.vue';
 import Custom from './custom.vue';
 import Iconfont from './iconfont.vue';
diff --git a/components/tree/demo/index.vue b/components/tree/demo/index.vue
index 494134cdc..ee14de182 100644
--- a/components/tree/demo/index.vue
+++ b/components/tree/demo/index.vue
@@ -24,7 +24,7 @@ import Directory from './directory.vue';
 import SwitcherIcon from './switcher-icon.vue';
 import ReplaceFields from './replaceFields.vue';
 import ContextMenu from './context-menu.vue';
-import VirtualScroll from './virtual-scroll';
+import VirtualScroll from './virtual-scroll.vue';
 import CN from '../index.zh-CN.md';
 import US from '../index.en-US.md';
 import { defineComponent } from 'vue';
diff --git a/package.json b/package.json
index dfb4d04c6..4650f3492 100644
--- a/package.json
+++ b/package.json
@@ -51,7 +51,8 @@
     "lint:style": "stylelint \"{site,components}/**/*.less\" --syntax less",
     "codecov": "codecov",
     "routes": "node site/scripts/genrateRoutes.js",
-    "tsc": "tsc --noEmit"
+    "tsc": "tsc --noEmit",
+    "site": "yarn routes && ./node_modules/vite/bin/vite.js build site --base=https://aliyuncdn.antdv.com/v2/"
   },
   "repository": {
     "type": "git",
diff --git a/site/themeConfig.ts b/site/themeConfig.ts
index 6ec49e3e1..6c38a978a 100644
--- a/site/themeConfig.ts
+++ b/site/themeConfig.ts
@@ -6,6 +6,7 @@ const themeConfig = [
     theme: 'dark',
     htmlThemeAttr: 'dark',
     modifyVars: {
+      hack: `true;@import "${require.resolve('../components/style/color/colorPalette.less')}";`,
       ...defaultVars,
       ...dark,
       'text-color': 'fade(@white, 65%)',
@@ -36,6 +37,8 @@ const additionalData = async (content: string, filename: string): Promise<string
       }
       return Promise.resolve(res);
     } catch (error) {
+      // eslint-disable-next-line no-console
+      console.log(error);
       return Promise.reject(content);
     }
   });
diff --git a/site/vite.config.ts b/site/vite.config.ts
index f95a90da4..6a3a643f1 100644
--- a/site/vite.config.ts
+++ b/site/vite.config.ts
@@ -13,6 +13,7 @@ export default {
     alias: {
       '@': path.join(__dirname, './src'),
       vue: 'vue/dist/vue.esm-bundler.js',
+      'ant-design-vue/es': path.resolve(__dirname, '../components'),
       'ant-design-vue': path.resolve(__dirname, '../components'),
     },
   },
@@ -43,7 +44,10 @@ export default {
   css: {
     preprocessorOptions: {
       less: {
-        modifyVars: { ...defaultVar },
+        modifyVars: {
+          hack: `true;@import "${require.resolve('../components/style/color/colorPalette.less')}";`,
+          ...defaultVar,
+        },
         javascriptEnabled: true,
         // includePaths: ["node_modules/"],
         additionalData,