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 a1cbc77b7a5..be4cd66353c 100644
--- a/tests/test_litellm/proxy/management_endpoints/test_team_endpoints.py
+++ b/tests/test_litellm/proxy/management_endpoints/test_team_endpoints.py
@@ -931,9 +931,10 @@ async def test_add_team_member_budget_table_success():
         team_info_response_object=team_info_response,
     )
 
-    # Verify the result
-    assert result == team_info_response
+    assert result.team_id == team_info_response.team_id
+    assert result.team_alias == team_info_response.team_alias
     assert result.team_member_budget_table == mock_budget_record
+    assert team_info_response.team_member_budget_table is None
 
     # Verify database call was made correctly
     mock_prisma_client.db.litellm_budgettable.find_unique.assert_called_once_with(