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 4ccfca14f4..42873479ca 100644
--- a/tests/test_litellm/proxy/management_endpoints/test_ui_sso.py
+++ b/tests/test_litellm/proxy/management_endpoints/test_ui_sso.py
@@ -1275,7 +1275,7 @@ class TestCLIKeyRegenerationFlow:
             )
             
             # Assert
-            mock_regenerate.assert_called_once_with(existing_key, new_key)
+            mock_regenerate.assert_called_once_with(existing_key=existing_key, new_key=new_key, user_id=None)
             assert result.status_code == 200
             assert "Success" in result.body.decode()