update tests to reflect current code