diff --git a/sumpy/kernel.py b/sumpy/kernel.py
index 768ad1a34ca6c9c8ce056ea499ed3e76e21a905a..3ce0bb8d3b654ca564ad672ef2417815be2a5eec 100644
--- a/sumpy/kernel.py
+++ b/sumpy/kernel.py
@@ -797,6 +797,9 @@ 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):
+        return type(self)(self.axis, new_inner_kernel)
+
     mapper_method = "map_axis_target_derivative"