Fix for change in API