[api2] Fix for api change