Fixed bug when returning to main menu