update tests to reflect changes
2 files changed