DI small refactoring