From 3115c64eb4feb11454a9c0470cb939a7e7bdd1cd Mon Sep 17 00:00:00 2001
From: Kevin Ingersoll <kingersoll@gmail.com>
Date: Mon, 15 Apr 2024 17:50:23 +0100
Subject: [PATCH] reset step after signing

---
 .../steps/app-account/AccountDelegationContent.tsx  | 13 +++++++++++--
 .../steps/app-account/useSignRegisterDelegation.ts  |  1 +
 2 files changed, 12 insertions(+), 2 deletions(-)

diff --git a/packages/account-kit/src/steps/app-account/AccountDelegationContent.tsx b/packages/account-kit/src/steps/app-account/AccountDelegationContent.tsx
index 4d09cafedf..8a4d01c6aa 100644
--- a/packages/account-kit/src/steps/app-account/AccountDelegationContent.tsx
+++ b/packages/account-kit/src/steps/app-account/AccountDelegationContent.tsx
@@ -3,9 +3,11 @@ import { AccountModalContent } from "../../AccountModalContent";
 import { AccountModalTitle } from "../../AccoutModalTitle";
 import { AppInfo } from "./AppInfo";
 import { useSignRegisterDelegation } from "./useSignRegisterDelegation";
+import { useOnboardingSteps } from "../../useOnboardingSteps";
 
 export function AccountDelegationContent() {
-  const { signRegisterDelegation, isPending, error } = useSignRegisterDelegation();
+  const { signRegisterDelegationAsync, isPending, error } = useSignRegisterDelegation();
+  const { resetStep } = useOnboardingSteps();
 
   return (
     <>
@@ -36,7 +38,14 @@ export function AccountDelegationContent() {
             for this app, and creating a signing key for a frictionless experience.
           </p>
 
-          <Button className="self-stretch" pending={isPending} onClick={() => signRegisterDelegation()}>
+          <Button
+            className="self-stretch"
+            pending={isPending}
+            onClick={async () => {
+              await signRegisterDelegationAsync();
+              resetStep();
+            }}
+          >
             Sign in
           </Button>
         </div>
diff --git a/packages/account-kit/src/steps/app-account/useSignRegisterDelegation.ts b/packages/account-kit/src/steps/app-account/useSignRegisterDelegation.ts
index eb37f6792c..1cc5509464 100644
--- a/packages/account-kit/src/steps/app-account/useSignRegisterDelegation.ts
+++ b/packages/account-kit/src/steps/app-account/useSignRegisterDelegation.ts
@@ -63,6 +63,7 @@ export function useSignRegisterDelegation() {
     () => ({
       ...result,
       signRegisterDelegation: result.mutate,
+      signRegisterDelegationAsync: result.mutateAsync,
       registerDelegationSignature,
     }),
     [registerDelegationSignature, result],