diff --git a/tests/libs/media_api/anilist/test_mapper.py b/tests/libs/media_api/anilist/test_mapper.py
index 0a20cf4..1aaa828 100644
--- a/tests/libs/media_api/anilist/test_mapper.py
+++ b/tests/libs/media_api/anilist/test_mapper.py
@@ -1,3 +1,5 @@
+from typing import Any
+
 from viu_media.libs.media_api.anilist.mapper import to_generic_user_profile
 from viu_media.libs.media_api.anilist.types import AnilistViewerData
 from viu_media.libs.media_api.types import UserProfile
@@ -9,8 +11,14 @@ def test_to_generic_user_profile_success():
             "Viewer": {
                 "id": 123,
                 "name": "testuser",
-                "avatar": {"large": "https://example.com/avatar.png"},
+                "avatar": {
+                    "large": "https://example.com/avatar.png",
+                    "medium": "https://example.com/avatar_medium.png",
+                    "extraLarge": "https://example.com/avatar_extraLarge.png",
+                    "small": "https://example.com/avatar_small.png",
+                },
                 "bannerImage": "https://example.com/banner.png",
+                "token": "test_token",
             }
         }
     }
@@ -23,24 +31,24 @@ def test_to_generic_user_profile_success():
 
 
 def test_to_generic_user_profile_data_none():
-    data = {"data": None}
+    data: Any = {"data": None}
     profile = to_generic_user_profile(data)
     assert profile is None
 
 
 def test_to_generic_user_profile_no_data_key():
-    data = {"errors": [{"message": "Invalid token"}]}
+    data: Any = {"errors": [{"message": "Invalid token"}]}
     profile = to_generic_user_profile(data)
     assert profile is None
 
 
 def test_to_generic_user_profile_no_viewer_key():
-    data: AnilistViewerData = {"data": {"Page": {}}}
+    data: Any = {"data": {"Page": {}}}
     profile = to_generic_user_profile(data)
     assert profile is None
 
 
 def test_to_generic_user_profile_viewer_none():
-    data: AnilistViewerData = {"data": {"Viewer": None}}
+    data: Any = {"data": {"Viewer": None}}
     profile = to_generic_user_profile(data)
     assert profile is None