Fix update to API