slight tweak to user flow