Fix endgame so that starting a new game works