diff --git a/packages/extension-polkagate/src/fullscreen/accountDetails/unlock/Review.tsx b/packages/extension-polkagate/src/fullscreen/accountDetails/unlock/Review.tsx index 194771e59..d01869e24 100644 --- a/packages/extension-polkagate/src/fullscreen/accountDetails/unlock/Review.tsx +++ b/packages/extension-polkagate/src/fullscreen/accountDetails/unlock/Review.tsx @@ -58,7 +58,7 @@ export default function Review ({ address, api, classToUnlock, setDisplayPopup, const [txInfo, setTxInfo] = useState(); const [estimatedFee, setEstimatedFee] = useState(); const [params, setParams] = useState[]>(); - const [step, setStep] = useState(STEPS.REVIEW); + const [step, setStep] = useState(STEPS.INDEX); const selectedProxyAddress = selectedProxy?.delegate as unknown as string; @@ -113,12 +113,13 @@ export default function Review ({ address, api, classToUnlock, setDisplayPopup, setDisplayPopup(undefined); }, [setDisplayPopup]); - const closeProxy = useCallback(() => setStep(STEPS.REVIEW), []); + const closeProxy = useCallback(() => setStep(STEPS.INDEX), []); return ( } - {[STEPS.REVIEW, STEPS.SIGN_QR].includes(step) && + {[STEPS.INDEX, STEPS.REVIEW, STEPS.SIGN_QR].includes(step) && <> diff --git a/packages/extension-polkagate/src/fullscreen/stake/solo/commonTasks/configurePayee/index.tsx b/packages/extension-polkagate/src/fullscreen/stake/solo/commonTasks/configurePayee/index.tsx index 5ef8850a0..013e5a30b 100644 --- a/packages/extension-polkagate/src/fullscreen/stake/solo/commonTasks/configurePayee/index.tsx +++ b/packages/extension-polkagate/src/fullscreen/stake/solo/commonTasks/configurePayee/index.tsx @@ -31,13 +31,17 @@ interface Props { setRefresh: React.Dispatch> } -export const ModalTitle = ({ icon, onCancel, setStep, step, text }: { text: string, onCancel: () => void, setStep?: React.Dispatch>, icon: IconDefinition, step: number }): React.ReactElement => { +export const ModalTitle = ({ closeProxy, icon, onCancel, setStep, step, text }: { closeProxy?: () => void, text: string, onCancel: () => void, setStep?: React.Dispatch>, icon: IconDefinition, step: number }): React.ReactElement => { const theme = useTheme(); const { t } = useTranslation(); - const closeProxy = useCallback( - () => setStep && setStep(STEPS.REVIEW) - , [setStep]); + const _closeProxy = useCallback(() => { + if (closeProxy) { + return closeProxy(); + } + + setStep && setStep(STEPS.REVIEW); + }, [closeProxy, setStep]); const onClose = useCallback(() => { setStep && setStep(STEPS.INDEX); @@ -65,7 +69,7 @@ export const ModalTitle = ({ icon, onCancel, setStep, step, text }: { text: stri step === STEPS.INDEX ? onCancel : step === STEPS.PROXY - ? closeProxy + ? _closeProxy : onClose } >