Can I undo the migration to non-custodial?

Migration to non-custodial mode should be treated as permanent. Once your funds are moved to your non-custodial wallet, you control them through that wallet and backup. In jurisdictions where custodial access is being phased out, returning to custodial mode may not be available.