Make it DI compatible