diff --git a/README.md b/README.md
index 8814b65a..18beadf2 100644
--- a/README.md
+++ b/README.md
@@ -1,4 +1,4 @@
-<div style="text-align:center">
+<div align="center">
 	<img src="https://i.loli.net/2021/11/24/x5lLfuSnEawBAgi.png"/>
 	<h1>Soybean Admin</h1>
 </div>
diff --git a/vite.config.ts b/vite.config.ts
index 429591c8..d23796b1 100644
--- a/vite.config.ts
+++ b/vite.config.ts
@@ -1,28 +1,32 @@
-import { defineConfig } from 'vite';
+import { defineConfig, loadEnv } from 'vite';
 import path from 'path';
-import { viteEnv, plugins, define } from './build';
+import { plugins, define } from './build';
 
-export default defineConfig({
-  base: viteEnv.VITE_BASE_URL,
-  resolve: {
-    alias: {
-      '@': path.resolve(__dirname, './src')
-    }
-  },
-  define,
-  css: {
-    preprocessorOptions: {
-      scss: {
-        additionalData: `@use "./src/styles/scss/global.scss" as *;`
+export default defineConfig(({ mode }) => {
+  const viteEnv = loadEnv(mode, `.env.${mode}`);
+
+  return {
+    base: viteEnv.VITE_BASE_URL,
+    resolve: {
+      alias: {
+        '@': path.resolve(__dirname, './src')
       }
+    },
+    define,
+    css: {
+      preprocessorOptions: {
+        scss: {
+          additionalData: `@use "./src/styles/scss/global.scss" as *;`
+        }
+      }
+    },
+    plugins,
+    server: {
+      host: '0.0.0.0'
+    },
+    build: {
+      brotliSize: false,
+      sourcemap: false
     }
-  },
-  plugins,
-  server: {
-    host: '0.0.0.0'
-  },
-  build: {
-    brotliSize: false,
-    sourcemap: false
-  }
+  };
 });