Minor fix in the check