diff --git a/src/frontend/vite.config.ts b/src/frontend/vite.config.ts
index c361b7ea4..d477ce539 100644
--- a/src/frontend/vite.config.ts
+++ b/src/frontend/vite.config.ts
@@ -5,7 +5,9 @@ const apiRoutes = ["^/api/v1/", "/health"];
 
 // Use environment variable to determine the target.
 const target = process.env.VITE_PROXY_TARGET || "http://127.0.0.1:7860";
-const port = process.env.VITE_PROXY_PORT || 3000;
+
+// Use environment variable to determine the UI server port 
+const port = process.env.VITE_PORT || 3000;
 
 const proxyTargets = apiRoutes.reduce((proxyObj, route) => {
   proxyObj[route] = {