diff --git a/sumpy/kernel.py b/sumpy/kernel.py
index 768ad1a34ca6c9c8ce056ea499ed3e76e21a905a..1ee007ada2bfc129c43fd6c5b59efb6a9d3a6476 100644
--- a/sumpy/kernel.py
+++ b/sumpy/kernel.py
@@ -797,6 +797,10 @@ class AxisTargetDerivative(DerivativeBase):
         expr = self.inner_kernel.postprocess_at_target(expr, bvec)
         return expr.diff(bvec[self.axis])
 
+    def replace_inner_kernel(self, new_inner_kernel):
+        self.inner_kernel = new_inner_kernel
+        return self
+
     mapper_method = "map_axis_target_derivative"