diff --git a/Controller/TokenController.php b/Controller/TokenController.php
index d5840276..6fb47ab8 100644
--- a/Controller/TokenController.php
+++ b/Controller/TokenController.php
@@ -13,7 +13,7 @@
 
 namespace FOS\OAuthServerBundle\Controller;
 
-use OAuth2\OAuth2;
+use OAuth2\IOAuth2;
 use OAuth2\OAuth2ServerException;
 use Symfony\Component\HttpFoundation\Request;
 use Symfony\Component\HttpFoundation\Response;
@@ -21,11 +21,11 @@
 class TokenController
 {
     /**
-     * @var OAuth2
+     * @var IOAuth2
      */
     protected $server;
 
-    public function __construct(OAuth2 $server)
+    public function __construct(IOAuth2 $server)
     {
         $this->server = $server;
     }