diff --git a/tests/test_litellm/proxy/management_endpoints/test_ui_sso.py b/tests/test_litellm/proxy/management_endpoints/test_ui_sso.py
index 3aaaebe0fb6..f7bf507c6da 100644
--- a/tests/test_litellm/proxy/management_endpoints/test_ui_sso.py
+++ b/tests/test_litellm/proxy/management_endpoints/test_ui_sso.py
@@ -2115,6 +2115,7 @@ class TestCLIKeyRegenerationFlow:
 
         # Mock request (no query params needed - existing_key is in state)
         mock_request = MagicMock(spec=Request)
+        mock_request.query_params = {}
 
         # CLI state with existing_key embedded: {PREFIX}:{key}:{existing_key}
         cli_state = f"{LITELLM_CLI_SESSION_TOKEN_PREFIX}:sk-new-session-key-456:sk-existing-cli-key-123"