diff --git a/tests/test_litellm/proxy/auth/test_handle_jwt.py b/tests/test_litellm/proxy/auth/test_handle_jwt.py
index cdb9ae4d9ab..a5764f40809 100644
--- a/tests/test_litellm/proxy/auth/test_handle_jwt.py
+++ b/tests/test_litellm/proxy/auth/test_handle_jwt.py
@@ -403,9 +403,9 @@ async def test_sync_user_role_and_teams_cache_invalidation_on_role_change():
     mock_cache.async_set_cache.assert_called_once()
     call_kwargs = mock_cache.async_set_cache.call_args
     assert call_kwargs.kwargs["key"] == "u1"
-    assert (
-        call_kwargs.kwargs["value"]["user_role"] == LitellmUserRoles.PROXY_ADMIN.value
-    )
+    assert isinstance(call_kwargs.kwargs["value"], LiteLLM_UserTable)
+    assert call_kwargs.kwargs["value"].user_role == LitellmUserRoles.PROXY_ADMIN.value
+    assert call_kwargs.kwargs["model_type"] == LiteLLM_UserTable
 
 
 @pytest.mark.asyncio
@@ -450,7 +450,9 @@ async def test_sync_user_role_and_teams_cache_invalidation_on_team_change():
     mock_cache.async_set_cache.assert_called_once()
     call_kwargs = mock_cache.async_set_cache.call_args
     assert call_kwargs.kwargs["key"] == "u1"
-    assert set(call_kwargs.kwargs["value"]["teams"]) == {"team1", "team2"}
+    assert isinstance(call_kwargs.kwargs["value"], LiteLLM_UserTable)
+    assert set(call_kwargs.kwargs["value"].teams) == {"team1", "team2"}
+    assert call_kwargs.kwargs["model_type"] == LiteLLM_UserTable
 
 
 @pytest.mark.asyncio