updating the tests to reflect the API change