bug fix on return to main menu from game