diff --git a/vite.config.js b/vite.config.js
index 0a4f25c..eb1f0ea 100644
--- a/vite.config.js
+++ b/vite.config.js
@@ -1,3 +1,4 @@
+import { resolve } from 'path';
 import { defineConfig } from 'vite';
 import { ViteEjsPlugin } from 'vite-plugin-ejs';
 import { viteSingleFile } from 'vite-plugin-singlefile';
@@ -31,7 +32,7 @@ export default defineConfig({
       {
         ejs: (viteConfig) => ({
           // ejs options goes here.
-          views: [__dirname],
+          views: [resolve(__dirname)],
         })
       }
     ),
diff --git a/vite.config.site.js b/vite.config.site.js
index 10d818c..11160b0 100644
--- a/vite.config.site.js
+++ b/vite.config.site.js
@@ -1,7 +1,10 @@
+import { resolve } from 'path';
 import { defineConfig } from 'vite';
 
 import { TransformEjs } from './src/lib/vite-plugins';
 
+const rootDir = resolve(__dirname, 'src')
+
 export default defineConfig({
   root: 'src',
   build: {
@@ -9,12 +12,9 @@ export default defineConfig({
     target: 'esnext',
     rollupOptions: {
       input: {
-        main: 'index.html',
-        editor: 'editor/index.html',
-        preview: 'preview/index.html',
-        // main: resolve(rootDir, 'index.html'),
-        // editor: resolve(rootDir, 'editor/index.html'),
-        // editorPreview: resolve(rootDir, 'editor/preview.html'),
+        main: resolve(rootDir, 'index.html'),
+        editor: resolve(rootDir, 'editor/index.html'),
+        preview: resolve(rootDir, 'preview/index.html'),
       },
     },
   },