forgot to update tests from refactor