update tests to reflect changes to api