Update according to API change