made some changes to run tests.