Updated the dependency injection.