diff --git a/src/main/java/vercel/VercelMPIdentityProvider.java b/src/main/java/vercel/VercelMPIdentityProvider.java
index df8da5c..badefd8 100755
--- a/src/main/java/vercel/VercelMPIdentityProvider.java
+++ b/src/main/java/vercel/VercelMPIdentityProvider.java
@@ -230,6 +230,7 @@ public Response authResponse(@QueryParam(AbstractOAuth2IdentityProvider.OAUTH2_P
                 authSession.setClientNote(OIDCLoginProtocol.REDIRECT_URI_PARAM, redirectUri);
                 authSession.setClientNote(OIDCLoginProtocol.RESPONSE_TYPE_PARAM, OAuth2Constants.CODE);
                 authSession.setClientNote(OIDCLoginProtocol.ISSUER, Urls.realmIssuer(uriInfo.getBaseUri(), realm.getName()));
+                authSession.setClientNote(OIDCLoginProtocol.STATE_PARAM, state);
 
                 session.getContext().setAuthenticationSession(authSession);