diff --git a/tests/local_testing/test_provider_budgets.py b/tests/local_testing/test_provider_budgets.py
index a26302e6d..17371da54 100644
--- a/tests/local_testing/test_provider_budgets.py
+++ b/tests/local_testing/test_provider_budgets.py
@@ -60,6 +60,9 @@ async def test_provider_budgets_e2e_test():
                 "model_info": {"id": "openai-model-id"},
             },
         ],
+        redis_host=os.getenv("REDIS_HOST"),
+        redis_port=int(os.getenv("REDIS_PORT")),
+        redis_password=os.getenv("REDIS_PASSWORD"),
     )
 
     response = await router.acompletion(
@@ -106,6 +109,9 @@ async def test_provider_budgets_e2e_test_expect_to_fail():
                 },
             },
         ],
+        redis_host=os.getenv("REDIS_HOST"),
+        redis_port=int(os.getenv("REDIS_PORT")),
+        redis_password=os.getenv("REDIS_PASSWORD"),
     )
 
     response = await router.acompletion(