Small fix in library