update tests to reflect changes