Minor change to end game logic.