Tweak to the change to logging out