diff --git a/frontend/vite.config.ts b/frontend/vite.config.ts
index 928304b..2baa4a4 100644
--- a/frontend/vite.config.ts
+++ b/frontend/vite.config.ts
@@ -118,7 +118,7 @@ export default defineConfig(({ mode }): UserConfig => ({
       '/api/prediction': {
         target: 'http://localhost:8001',
         changeOrigin: true,
-        rewrite: (path) => path.replace(/^\/api\/prediction/, ''),
+        rewrite: (path) => path.replace(/^\/api\/prediction/, '/api'),
       },
       '/ollama': {
         target: 'http://localhost:11434',