Fixed the API slightly.