I have done that before.
I’d be satisfied just to have a sensible and consistent back button experience.