diff --git a/app/Model/UserSite.php b/app/Model/UserSite.php
index 65f8fae..1388028 100644
--- a/app/Model/UserSite.php
+++ b/app/Model/UserSite.php
@@ -51,9 +51,9 @@ class UserSite extends Model
     /**
      * @param int $userId
      * @param int $siteId
-     * @return Collection
+     * @return Builder|\Hyperf\Database\Model\Model|null
      */
-    public function getInfoByUserIdAndSiteId(int $userId, int $siteId): Collection
+    public function getInfoByUserIdAndSiteId(int $userId, int $siteId): \Hyperf\Database\Model\Model|Builder|null
     {
         return $this->where('user_id', $userId)->where('site_id', $siteId)->first();
     }