Fix stalemate message showing after flag has been returned/captured