update tests to match