Fix bug in DI Container generation