Fix the type in the API