Minor update to the previous handling of ending a game