Updated map to work with changes in the API.