Fix endgame bug