Fix DependencyInjection