diff --git a/package.json b/package.json
index 507af58..63acc43 100644
--- a/package.json
+++ b/package.json
@@ -4,6 +4,7 @@
   "description": "Electron React Vite boilerplate.",
   "author": "草鞋没号 <308487730@qq.com>",
   "license": "MIT",
+  "main": "dist/main/index.js",
   "scripts": {
     "dev": "node -r ts-node/register scripts/dev.ts",
     "build": "node -r ts-node/register scripts/build.ts"
@@ -32,6 +33,7 @@
     "vite": "^2.6.13"
   },
   "env": {
+    "HOST": "127.0.0.1",
     "PORT": 3344,
     "PORT_WS": 3355
   }
diff --git a/react-ts/vite.config.ts b/react-ts/vite.config.ts
index 39a76d7..b447ba9 100644
--- a/react-ts/vite.config.ts
+++ b/react-ts/vite.config.ts
@@ -1,6 +1,7 @@
 import { join } from 'path'
 import { defineConfig } from 'vite'
 import react from '@vitejs/plugin-react'
+import pkg from '../package.json'
 
 // https://vitejs.dev/config/
 export default defineConfig({
@@ -10,4 +11,8 @@ export default defineConfig({
     minify: false,
     outDir: join(process.cwd(), 'dist/react-ts'),
   },
+  server: {
+    host: pkg.env.HOST,
+    port: pkg.env.PORT,
+  },
 })