Fix bug in comments logic.