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 4c99f154d7b..d198b047d91 100644
--- a/tests/test_litellm/proxy/management_endpoints/test_team_endpoints.py
+++ b/tests/test_litellm/proxy/management_endpoints/test_team_endpoints.py
@@ -14224,6 +14224,4 @@ async def test_team_info_returns_parent_organization_models(organization, expect
             user_api_key_dict=UserAPIKeyAuth(user_role=LitellmUserRoles.PROXY_ADMIN),
         )
 
-    team_info = response["team_info"]
-    assert team_info.organization_models == expected_models
-    assert "litellm_organization_table" not in team_info.model_dump()
+    assert response["team_info"].organization_models == expected_models