diff --git a/api/libs/helper.py b/api/libs/helper.py
index 023240a9a4..b98a4829e8 100644
--- a/api/libs/helper.py
+++ b/api/libs/helper.py
@@ -31,12 +31,12 @@ class AppIconUrlField(fields.Raw):
         if obj is None:
             return None
 
-        from models.model import App, IconType
+        from models.model import App, IconType, Site
 
         if isinstance(obj, dict) and "app" in obj:
             obj = obj["app"]
 
-        if isinstance(obj, App) and obj.icon_type == IconType.IMAGE.value:
+        if isinstance(obj, App | Site) and obj.icon_type == IconType.IMAGE.value:
             return file_helpers.get_signed_file_url(obj.icon)
         return None