Fixed dependency injection for extra class