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) {