fix a build break due to api change