Minor fix in mlllib