Fixed the DI