Dependency Injection

Fixed class name of extension