Update for Dependency Injection example.