Fix bug when returning to game from menu