@@ -2,8 +2,8 @@ import React, { useCallback, useMemo } from 'react';
22import cn from 'classnames';
33import { Grid, H3, B, Text, CheckboxInput } from 'indigo-react';
44
5−import { useHistory } from 'store/history';
65import useHasDisclaimed from 'lib/useHasDisclaimed';
6+import { useLocalRouter } from 'lib/LocalRouter';
77
88import View from 'components/View';
99import WarningBox from 'components/WarningBox';
@@ -14,7 +14,7 @@ import { composeValidator, buildCheckboxValidator } from 'form/validators';
1414const TEXT_STYLE = 'f5';
1515
1616export default function ActivateDisclaimer() {
17− const { pop } = useHistory();
17+ const { pop, popAndPush, data } = useLocalRouter();
1818 const [, setHasDisclaimed] = useHasDisclaimed();
1919
2020 const validate = useMemo(
@@ -24,10 +24,14 @@ export default function ActivateDisclaimer() {
2424
2525 const initialValues = useMemo(() => ({ checkbox: false }), []);
2626
27− const goBack = useCallback(async () => {
27+ const goNext = useCallback(async () => {
2828 setHasDisclaimed(true);
29− pop();
30− }, [pop, setHasDisclaimed]);
29+ if (data.next) {
30+ popAndPush(data.next);
31+ } else {
32+ pop();
33+ }
34+ }, [data, pop, popAndPush, setHasDisclaimed]);
3135
3236 return (
3337 <View>
@@ -81,7 +85,7 @@ export default function ActivateDisclaimer() {
8185 </Grid.Item>
8286 <BridgeForm
8387 validate={validate}
84− afterSubmit={goBack}
88+ afterSubmit={goNext}
8589 initialValues={initialValues}>
8690 {({ handleSubmit }) => (
8791 <>