Rewrite dependency injection section