Small fix to lib