Allow to set `codeobj_class` for custom operations