fixing tests broken from refactor