Fix the back bug

Fix the bug when click on the back button