Fixed the API, oops