diff --git a/src/TokenType/TokenTypeInterface.php b/src/TokenType/TokenTypeInterface.php
index 610e9f16..1757a07b 100644
--- a/src/TokenType/TokenTypeInterface.php
+++ b/src/TokenType/TokenTypeInterface.php
@@ -12,6 +12,7 @@
 namespace League\OAuth2\Server\TokenType;
 
 use League\OAuth2\Server\AbstractServer;
+use League\OAuth2\Server\Entity\SessionEntity;
 use Symfony\Component\HttpFoundation\Request;
 
 interface TokenTypeInterface
@@ -43,6 +44,12 @@ interface TokenTypeInterface
      */
     public function getParam($key);
 
+    /**
+     * @param \League\OAuth2\Server\Entity\SessionEntity $session
+     * @return self
+     */
+    public function setSession(SessionEntity $session);
+
     /**
      * Determine the access token in the authorization header
      * @param  \Symfony\Component\HttpFoundation\Request $request