diff --git a/demo/loaders/ComponentDocumentationEntryTemplate.vue b/demo/loaders/ComponentDocumentationEntryTemplate.vue
index cff05e3fd..8bd0e4d1c 100644
--- a/demo/loaders/ComponentDocumentationEntryTemplate.vue
+++ b/demo/loaders/ComponentDocumentationEntryTemplate.vue
@@ -8,8 +8,10 @@
 <script>
 import zhCN from './zhCN/index'
 import enUS from './enUS/index'
+
 zhCN.name = 'zh-CN'
 enUS.name = 'en-US'
+
 export default {
   components: {
     'en-US': enUS,
diff --git a/demo/loaders/NaiveUIDocEntryLoader.js b/demo/loaders/NaiveUIDocEntryLoader.js
index 8f791e06c..ff7a9e4be 100644
--- a/demo/loaders/NaiveUIDocEntryLoader.js
+++ b/demo/loaders/NaiveUIDocEntryLoader.js
@@ -2,6 +2,17 @@ const path = require('path')
 const fs = require('fs')
 const template = fs.readFileSync(path.join(__dirname, 'ComponentDocumentationEntryTemplate.vue')).toString()
 
+// user may use different index for a locale
+// for example
+// enUS/index.md
+// zhCN/index.md
+//
+// enUS/index.demo-entry.md
+// zhCN/index.demo-entry.md
+//
+// enUS/index.vue
+// zhCN/index.vue
+
 module.exports = (entryPath) => {
   let resultTemplate = template
   const replaceReg = /index/g
diff --git a/demo/loaders/convertMd2Demo.js b/demo/loaders/convertMd2Demo.js
index 7758ba87d..55745d12a 100644
--- a/demo/loaders/convertMd2Demo.js
+++ b/demo/loaders/convertMd2Demo.js
@@ -95,6 +95,8 @@ function genVueComponent (parts, fileName, relativeUrl, noRunning = false) {
   }
   if (parts.script && !noRunning) {
     src = src.replace(scriptReg, parts.script)
+  } else {
+    src = src.replace(scriptReg, 'export default {}')
   }
   if (parts.style) {
     src = src.replace(styleReg, parts.style)
diff --git a/demo/vite-plugins/mdPlugin.js b/demo/vite-plugins/mdPlugin.js
index 6aaff3f00..c495c7c96 100644
--- a/demo/vite-plugins/mdPlugin.js
+++ b/demo/vite-plugins/mdPlugin.js
@@ -42,18 +42,19 @@ module.exports = {
   },
   rollupMdPlugin: function () {
     return {
-      async transform (code, id) {
-        console.log('id', id)
+      async load (id) {
         if (id.endsWith('.demo.md')) {
+          const code = readFileSync(id).toString()
           return NaiveUIDemoLoader(code, id)
-        } else if (id.endsWith('.demo-entry.md')) {
-          return NaiveUIDocEntryLoader(id)
         } else if (id.endsWith('.entry')) {
-          NaiveUIDocLoader(code, id)
+          return NaiveUIDocEntryLoader(id)
+        } else if (id.endsWith('.demo-entry.md')) {
+          const code = readFileSync(id).toString()
+          return NaiveUIDocLoader(code, id)
         } else if (id.endsWith('.md')) {
+          const code = readFileSync(id).toString()
           return NaiveUIMdLoader(code, id)
         }
-        return code
       }
     }
   }
diff --git a/rollup.config.js b/rollup.config.js
index ddda84841..d4c99a163 100644
--- a/rollup.config.js
+++ b/rollup.config.js
@@ -21,11 +21,7 @@ module.exports = {
     nodeResolve({
       extensions: ['.js', '.json', '.vue']
     }),
-    vuePlugin({
-      compilerOptions: {
-        whitespace: 'condense'
-      }
-    }),
+    vuePlugin(),
     terser({
       mangle: false,
       output: {
diff --git a/vite.config.js b/vite.config.js
index ff16351dc..9d8d48c46 100644
--- a/vite.config.js
+++ b/vite.config.js
@@ -5,7 +5,9 @@ const { viteMdPlugin, rollupMdPlugin } = require('./demo/vite-plugins/mdPlugin')
 
 module.exports = {
   root: __dirname,
-  plugins: [ viteMdPlugin() ],
+  plugins: [
+    viteMdPlugin()
+  ],
   optimizeDeps: {
     include: [
       'highlight.js/lib/languages/cpp',
@@ -24,6 +26,9 @@ module.exports = {
     'process.env.NODE_ENV': `'${process.env.NODE_ENV}'`,
     '__DEV__': process.env !== 'production'
   },
+  rollupPluginVueOptions: {
+    include: /\.(vue|md|demo.md|demo-entry.md|entry)$/
+  },
   rollupInputOptions: {
     plugins: [
       alias({
@@ -31,6 +36,10 @@ module.exports = {
           {
             find: /^naive-ui\/lib\/icons\//g,
             replacement: './src/_deprecated/icons/'
+          },
+          {
+            find: /^naive-ui$/g,
+            replacement: './src/index.js'
           }
         ]
       }),