updating for the API change