DI fixes