Update the API

Fix API