Fixed bug with DI.