diff --git a/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application/Services/Chat/AiImageService.cs b/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application/Services/Chat/AiImageService.cs
index 4e4c44a9..fc1487e9 100644
--- a/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application/Services/Chat/AiImageService.cs
+++ b/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application/Services/Chat/AiImageService.cs
@@ -348,7 +348,7 @@ public class AiImageService : ApplicationService
                 ModelName = x.Name,
                 ModelDescribe = x.Description,
                 Remark = x.Description,
-                IsPremiumPackage = PremiumPackageConst.ModeIds.Contains(x.ModelId)
+                IsPremiumPackage = x.IsPremium
             }).ToListAsync();
         return output;
     }