updated to API change