update tests to reflect changes in #281