update tests to reflect the changes