diff --git a/tests/test_litellm/proxy/management_endpoints/test_team_endpoints.py b/tests/test_litellm/proxy/management_endpoints/test_team_endpoints.py
index 5f3974b46fb..0bc1132be9e 100644
--- a/tests/test_litellm/proxy/management_endpoints/test_team_endpoints.py
+++ b/tests/test_litellm/proxy/management_endpoints/test_team_endpoints.py
@@ -4025,6 +4025,7 @@ async def test_new_team_max_budget_within_user_limit():
         mock_prisma.db.litellm_usertable = MagicMock()
         mock_prisma.db.litellm_usertable.upsert = AsyncMock(return_value=mock_user)
         mock_prisma.db.litellm_usertable.update = AsyncMock(return_value=mock_user)
+        mock_prisma.db.execute_raw = AsyncMock()
 
         # Mock team membership table
         mock_membership = MagicMock()
@@ -4166,6 +4167,7 @@ async def test_new_team_org_scoped_budget_bypasses_user_limit():
         mock_prisma.db.litellm_usertable = MagicMock()
         mock_prisma.db.litellm_usertable.upsert = AsyncMock(return_value=mock_user)
         mock_prisma.db.litellm_usertable.update = AsyncMock(return_value=mock_user)
+        mock_prisma.db.execute_raw = AsyncMock()
 
         # Mock team membership table
         mock_membership = MagicMock()
@@ -4312,6 +4314,7 @@ async def test_new_team_org_scoped_models_bypasses_user_limit():
         mock_prisma.db.litellm_usertable = MagicMock()
         mock_prisma.db.litellm_usertable.upsert = AsyncMock(return_value=mock_user)
         mock_prisma.db.litellm_usertable.update = AsyncMock(return_value=mock_user)
+        mock_prisma.db.execute_raw = AsyncMock()
 
         # Mock team membership table
         mock_membership = MagicMock()
@@ -7164,6 +7167,7 @@ async def test_new_team_soft_budget_validation(
         mock_prisma.db.litellm_usertable = MagicMock()
         mock_prisma.db.litellm_usertable.upsert = AsyncMock(return_value=mock_user)
         mock_prisma.db.litellm_usertable.update = AsyncMock(return_value=mock_user)
+        mock_prisma.db.execute_raw = AsyncMock()
 
         # Mock team membership table
         mock_membership = MagicMock()