diff --git a/src/Traits/HasRoles.php b/src/Traits/HasRoles.php
index 8f3f874..ab3fc49 100644
--- a/src/Traits/HasRoles.php
+++ b/src/Traits/HasRoles.php
@@ -90,7 +90,7 @@ public function scopePermission(Builder $query, $permissions): Builder
      *
      * @param array|string|Role ...$roles
      *
-     * @return $this
+     * @return array|Role|string
      */
     public function assignRole(...$roles)
     {
@@ -108,7 +108,7 @@ public function assignRole(...$roles)
 
         $this->forgetCachedPermissions();
 
-        return $this;
+        return $roles;
     }
 
     /**
@@ -116,7 +116,7 @@ public function assignRole(...$roles)
      *
      * @param array|string|Role ...$roles
      *
-     * @return HasRoles
+     * @return array|Role|string
      */
     public function removeRole(...$roles)
     {
@@ -131,7 +131,7 @@ public function removeRole(...$roles)
 
         $this->forgetCachedPermissions();
 
-        return $this;
+        return $roles;
     }
 
     /**