Reflect api change in new version