Minor fix in library