Reflect the API update

[ci skip]