Getting DI working with tests