More DI simplification