diff --git a/litellm/proxy/management_endpoints/key_management_endpoints.py b/litellm/proxy/management_endpoints/key_management_endpoints.py
index db0f35b6ffe..cd9dabbc34f 100644
--- a/litellm/proxy/management_endpoints/key_management_endpoints.py
+++ b/litellm/proxy/management_endpoints/key_management_endpoints.py
@@ -1308,6 +1308,7 @@ async def generate_key_helper_fn(  # noqa: PLR0915
     object_permission_id: Optional[
         str
     ] = None,  # object_permission_id <-> LiteLLM_ObjectPermissionTable
+    object_permission: Optional[LiteLLM_ObjectPermissionBase] = None,
 ):
     from litellm.proxy.proxy_server import (
         litellm_proxy_budget_name,