Corrected DI bug