diff --git a/devbox.json b/devbox.json
index 1adab88e..18c3f096 100644
--- a/devbox.json
+++ b/devbox.json
@@ -22,7 +22,7 @@
   "shell": {
     "init_hook": [
       "export DB_URI=postgres://${DB_USER}:${DB_PASS}@${DB_HOST}:${DB_PORT}/${DB_NAME}?sslmode=disable",
-      "export DB_URI_TEST=postgres://${DB_USER}:${DB_PASS}@${DB_HOST}:${DB_PORT_TEST}/${DB_NAME}?sslmode=disable",
+      "export DB_URI_TEST=postgres://${DB_USER}:${DB_PASS}@${DB_HOST}:${DB_PORT_GENERATE}/${DB_NAME}?sslmode=disable",
       "echo 'Welcome to the DoczyAI devbox!'"
     ]
   },
@@ -32,7 +32,7 @@
     "DB_PASS":               "pass",
     "DB_HOST":               "localhost",
     "DB_PORT":               "5432",
-    "DB_PORT_TEST":               "5430",
+    "DB_PORT_GENERATE":               "5431",
     "DB_NAME":               "query_orchestration",
     "DB_NOSSL":              "true",
     "AWS_ACCESS_KEY_ID":     "test",