diff --git a/src/pystencils_autodiff/_autodiff.py b/src/pystencils_autodiff/_autodiff.py
index 7735dccbb357eadd23a5882d832f832b38731838..4edd071b606b6c733046306b85e048b992e37696 100644
--- a/src/pystencils_autodiff/_autodiff.py
+++ b/src/pystencils_autodiff/_autodiff.py
@@ -62,6 +62,10 @@ Backward:
         self._time_constant_fields = time_constant_fields
         self._kwargs = kwargs
         self.op_name = op_name
+        self._forward_ast_cpu = None
+        self._backward_ast_cpu = None
+        self._forward_ast_gpu = None
+        self._backward_ast_gpu = None
         self._forward_kernel_cpu = None
         self._backward_kernel_cpu = None
         self._forward_kernel_gpu = None