diff --git a/src/IdTokenResponse.php b/src/IdTokenResponse.php
index 2a24e68..29e1d18 100644
--- a/src/IdTokenResponse.php
+++ b/src/IdTokenResponse.php
@@ -50,14 +50,12 @@ protected function getBuilder(AccessTokenEntityInterface $accessToken, UserEntit
         }
 
         // Add required id_token claims
-        $builder
+        return $builder
             ->permittedFor($accessToken->getClient()->getIdentifier())
             ->issuedBy('https://' . $_SERVER['HTTP_HOST'])
             ->issuedAt(new \DateTimeImmutable())
             ->expiresAt($expiresAt)
             ->relatedTo($userEntity->getIdentifier());
-
-        return $builder;
     }
 
     /**