[PATCH] Update the API code