bug correction in DI.