Fix bug for the win case