Fix for using latest api.