Slight mod on API.