Small fix comments logic