updated for running tests