diff --git a/litellm/proxy/management_endpoints/team_endpoints.py b/litellm/proxy/management_endpoints/team_endpoints.py
index a1b1c618f..5beafc13e 100644
--- a/litellm/proxy/management_endpoints/team_endpoints.py
+++ b/litellm/proxy/management_endpoints/team_endpoints.py
@@ -571,6 +571,11 @@ async def team_member_add(
 
         await asyncio.gather(*tasks)
 
+    # Check if updated_team is None
+    if updated_team is None:
+        raise HTTPException(
+            status_code=404, detail={"error": f"Team with id {data.team_id} not found"}
+        )
     return TeamAddMemberResponse(
         **updated_team.model_dump(),
         updated_users=updated_users,
diff --git a/litellm/proxy/management_helpers/utils.py b/litellm/proxy/management_helpers/utils.py
index b7ba5278b..a361a442f 100644
--- a/litellm/proxy/management_helpers/utils.py
+++ b/litellm/proxy/management_helpers/utils.py
@@ -103,7 +103,7 @@ async def add_new_member(
         elif len(existing_user_row) == 1:
             user_info = existing_user_row[0]
             _returned_user = await prisma_client.db.litellm_usertable.update(
-                where={"user_id": user_info.user_id},
+                where={"user_id": user_info.user_id},  # type: ignore
                 data={"teams": {"push": [team_id]}},
             )
             returned_user = LiteLLM_UserTable(**_returned_user.model_dump())