Fix screen sometimes being reset back to the main menu.