needed to refactor for the testing framework