diff --git a/vite.config.ts b/vite.config.ts
index 6c70beb..07318c9 100644
--- a/vite.config.ts
+++ b/vite.config.ts
@@ -1,27 +1,11 @@
 import react from "@vitejs/plugin-react";
 import { defineConfig } from "vite";
 
-const allowedHosts = [".develop.taylordb.ai", "localhost", "127.0.0.1"];
-
-// https://vitejs.dev/config/
+// https://vite.dev/config/
 export default defineConfig({
-  plugins: [
-    react(),
-    {
-      name: "host-validation",
-      configureServer(server) {
-        server.middlewares.use((req, res, next) => {
-          const host = req.headers.host?.split(":")[0];
-          if (host && allowedHosts.some((allowed) => host.endsWith(allowed))) {
-            return next();
-          }
-          res.writeHead(403, { "Content-Type": "text/plain" });
-          res.end(`Forbidden host: ${host}`);
-        });
-      },
-    },
-  ],
+  plugins: [react()],
   server: {
+    allowedHosts: [".develop.taylordb.ai", "localhost", "127.0.0.1"],
     host: "0.0.0.0", // Listen on all network interfaces
   },
 });