Updated tests to use correct API response format