Added in missing class for injection.