Fixed the DI files