update tests, changes to api