DI implementation fix