DI behaviour refactoring.