fixed an issue with DI