Update test to match last changes in the API