Dependency Injection for Edit Mode