Minor fix to handle changes to the API.