Fix view after going back to menu