fixed tests to use DI