Fix key after api change