fix the back button bug