diff --git a/litellm/proxy/_types.py b/litellm/proxy/_types.py
index 68d0554c5f8..bb7c8997af4 100644
--- a/litellm/proxy/_types.py
+++ b/litellm/proxy/_types.py
@@ -1752,7 +1752,7 @@ class NewTeamRequest(TeamBase):
     team_member_key_duration: Optional[str] = None  # e.g. "1d", "1w", "1m"
     team_member_budget_duration: Optional[str] = None  # e.g. "30d", "1mo"
     team_member_model_max_budget: Optional[dict] = (
-        None  # e.g. {"gpt-4": {"max_budget": 10.0, "budget_duration": "30d"}}
+        None  # e.g. {"gpt-4": {"max_budget": 10.0}}
     )
     allowed_vector_store_indexes: Optional[List[AllowedVectorStoreIndexItem]] = None
     enforced_batch_output_expires_after: Optional[dict] = None
@@ -1823,7 +1823,7 @@ class UpdateTeamRequest(LiteLLMPydanticObjectBase):
         None  # default allowed_models seeded onto new team members
     )
     team_member_model_max_budget: Optional[dict] = (
-        None  # e.g. {"gpt-4": {"max_budget": 10.0, "budget_duration": "30d"}}
+        None  # e.g. {"gpt-4": {"max_budget": 10.0}}
     )