update tests to reflect change