diff --git a/tests/Grant/DeviceCodeGrantTest.php b/tests/Grant/DeviceCodeGrantTest.php
index 9e1359661..7b4e09f1a 100644
--- a/tests/Grant/DeviceCodeGrantTest.php
+++ b/tests/Grant/DeviceCodeGrantTest.php
@@ -368,9 +368,7 @@ public function testRespondToAccessTokenRequest(): void
         $deviceCodeEntity->setExpiryDateTime(new DateTimeImmutable('+1 hour'));
         $deviceCodeEntity->setClient($client);
 
-        $deviceCodeRepositoryMock->method('getDeviceCodeEntityByDeviceCode')
-            ->with('deviceCodeEntityIdentifier')
-            ->willReturn($deviceCodeEntity);
+        $deviceCodeRepositoryMock->method('getDeviceCodeEntityByDeviceCode')->willReturn($deviceCodeEntity);
 
         $scopeRepositoryMock = $this->getMockBuilder(ScopeRepositoryInterface::class)->getMock();
         $scopeRepositoryMock->method('getScopeEntityByIdentifier')->willReturn($scope);
@@ -390,7 +388,7 @@ public function testRespondToAccessTokenRequest(): void
         $grant->setEncryptionKey($this->cryptStub->getKey());
         $grant->setPrivateKey(new CryptKey('file://' . __DIR__ . '/../Stubs/private.key'));
 
-        $grant->completeDeviceAuthorizationRequest('deviceCodeEntityIdentifier', 'baz', true);
+        $grant->completeDeviceAuthorizationRequest($deviceCodeEntity->getIdentifier(), 'baz', true);
 
         $serverRequest = (new ServerRequest())->withParsedBody([
             'grant_type' => 'urn:ietf:params:oauth:grant-type:device_code',