From 3ccf53891cf9c26798d4648849f03a2be388b5cf Mon Sep 17 00:00:00 2001
From: Nuthara Andrahennedi <nuthara@wso2.com>
Date: Mon, 13 Jul 2026 17:16:48 +0530
Subject: [PATCH] Fix stale challenge token on redirection

---
 .../src/components/presentation/auth/SignUp/BaseSignUp.tsx      | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/packages/react/src/components/presentation/auth/SignUp/BaseSignUp.tsx b/packages/react/src/components/presentation/auth/SignUp/BaseSignUp.tsx
index 5b1476c..9c49e71 100644
--- a/packages/react/src/components/presentation/auth/SignUp/BaseSignUp.tsx
+++ b/packages/react/src/components/presentation/auth/SignUp/BaseSignUp.tsx
@@ -651,6 +651,7 @@ const BaseSignUpContent: FC<BaseSignUpProps> = ({
 
           try {
             const continueResponse: any = await onSubmit!(payload);
+            await setChallengeToken(continueResponse.challengeToken ?? null);
             onFlowChange?.(continueResponse);
 
             if (continueResponse.flowStatus === EmbeddedSignUpFlowStatus.Error) {
@@ -730,6 +731,7 @@ const BaseSignUpContent: FC<BaseSignUpProps> = ({
 
                 try {
                   const continueResponse: any = await onSubmit!(payload);
+                  await setChallengeToken(continueResponse.challengeToken ?? null);
                   onFlowChange?.(continueResponse);
 
                   if (continueResponse.flowStatus === EmbeddedSignUpFlowStatus.Error) {