- updated the test case to reflect the api changes.