Instead of having such comments, we should make the logic easier to understand.