Fix api break