Fix DI