diff --git a/apps/mobile/app/(app)/budget/[budgetId]/edit.tsx b/apps/mobile/app/(app)/budget/[budgetId]/edit.tsx index 64be2826..39ea97e8 100644 --- a/apps/mobile/app/(app)/budget/[budgetId]/edit.tsx +++ b/apps/mobile/app/(app)/budget/[budgetId]/edit.tsx @@ -22,7 +22,7 @@ export default function EditBudgetScreen() { const { i18n } = useLingui() const { budgetId } = useLocalSearchParams<{ budgetId: string }>() const { budget } = useBudget(budgetId!) - const { mutateAsync } = useUpdateBudget() + const { mutateAsync: updateUpdate } = useUpdateBudget() const { mutateAsync: mutateDelete } = useDeleteBudget() const { setDefaultBudgetId, defaultBudgetId } = useUserMetadata() const { sideOffset, ...rootProps } = useModalPortalRoot() @@ -69,9 +69,12 @@ export default function EditBudgetScreen() { const handleUpdate = async ({ isDefault, ...data }: BudgetFormValues) => { if (isDefault) { - await setDefaultBudgetId(budget?.id) + setDefaultBudgetId(budget?.id) + } else if (defaultBudgetId === budget?.id) { + // unset default budget + setDefaultBudgetId(undefined) } - mutateAsync({ + updateUpdate({ data: data, id: budget?.id!, }).catch(() => {