diff --git a/src/Storage/SessionInterface.php b/src/Storage/SessionInterface.php
index 0da92233..40f8873f 100644
--- a/src/Storage/SessionInterface.php
+++ b/src/Storage/SessionInterface.php
@@ -25,11 +25,18 @@ interface SessionInterface
 
     /**
      * Get a session from an access token
-     * @param  string $accessToken The access token
+     * @param  \League\OAuth2\Server\Entity\AccessToken $accessToken The access token
      * @return \League\OAuth2\Server\Entity\Session
      */
     public function getByAccessToken($accessToken);
 
+    /**
+     * Get a session from an auth code
+     * @param  \League\OAuth2\Server\Entity\AuthCode $authCode The auth code
+     * @return \League\OAuth2\Server\Entity\Session
+     */
+    public function getByAuthCode($authCode);
+
     /**
      * Get a session's scopes
      * @param  integer $sessionId