Fix bug for DI configuration