diff --git a/src/HasApiTokens.php b/src/HasApiTokens.php
index 71853fed0..025c7f2e1 100644
--- a/src/HasApiTokens.php
+++ b/src/HasApiTokens.php
@@ -71,7 +71,7 @@ public function createToken($name, array $scopes = [])
     /**
      * Set the current access token for the user.
      *
-     * @param  \Laravel\Passport\Token  $accessToken
+     * @param  \Laravel\Passport\Token|\Laravel\Passport\TransientToken  $accessToken
      * @return $this
      */
     public function withAccessToken($accessToken)