update tests to match implementation