update tests to reflect latest changes