diff --git a/tests/DispatcherTest.php b/tests/DispatcherTest.php
index 115931f5..87dd3d22 100644
--- a/tests/DispatcherTest.php
+++ b/tests/DispatcherTest.php
@@ -1,6 +1,7 @@
 <?php
 namespace Szurubooru\Tests;
 use Szurubooru\Dispatcher;
+use Szurubooru\Entities\Token;
 use Szurubooru\Helpers\HttpHelper;
 use Szurubooru\RouteRepository;
 use Szurubooru\Router;
@@ -66,7 +67,7 @@ final class DispatcherTest extends AbstractDatabaseTestCase
     public function testAuthorization()
     {
         $this->httpHelperMock->expects($this->once())->method('getRequestHeader')->with($this->equalTo('X-Authorization-Token'))->willReturn('test');
-        $this->tokenServiceMock->expects($this->once())->method('getByName');
+        $this->tokenServiceMock->expects($this->once())->method('getByName')->willReturn(new Token());
         $this->routeRepositoryMock->expects($this->once())->method('injectRoutes');
 
         $dispatcher = $this->getDispatcher();