update tests to match changes