Fix for slight change in API response