Fix bug in DI configuration