Fixed an issue with the change in API response