diff --git a/kernex/src/map.py b/kernex/src/map.py
index 26f0c88..ce913aa 100644
--- a/kernex/src/map.py
+++ b/kernex/src/map.py
@@ -54,7 +54,7 @@ def __multi_call__(self, array, *args, **kwargs):
         return result.reshape(*self.output_shape, *func_shape)
 
 
-@pytc.treeclass(op=False)
+@pytc.treeclass
 class kernelMap(baseKernelMap):
     def __init__(self, func_dict, shape, kernel_size, strides, padding, relative):
 
@@ -64,7 +64,7 @@ def __call__(self, array, *args, **kwargs):
         return self.__call__(array, *args, **kwargs)
 
 
-@pytc.treeclass(op=False)
+@pytc.treeclass
 class offsetKernelMap(kernelMap):
     def __init__(self, func_dict, shape, kernel_size, strides, offset, relative):
 
diff --git a/kernex/src/scan.py b/kernex/src/scan.py
index 95c250e..ab87646 100644
--- a/kernex/src/scan.py
+++ b/kernex/src/scan.py
@@ -62,7 +62,7 @@ def scan_body(padded_array, view):
         )
 
 
-@pytc.treeclass(op=False)
+@pytc.treeclass
 class kernelScan(baseKernelScan):
     def __init__(self, func_dict, shape, kernel_size, strides, padding, relative):
 
@@ -72,7 +72,7 @@ def __call__(self, array, *args, **kwargs):
         return self.__call__(array, *args, **kwargs)
 
 
-@pytc.treeclass(op=False)
+@pytc.treeclass
 class offsetKernelScan(kernelScan):
     def __init__(self, func_dict, shape, kernel_size, strides, offset, relative):