Minor fix to comments