Small fix in API