diff --git a/xdem/coreg/base.py b/xdem/coreg/base.py
index fea745d5..118c1516 100644
--- a/xdem/coreg/base.py
+++ b/xdem/coreg/base.py
@@ -717,7 +717,7 @@ def _postprocess_coreg_apply_pts(
     return applied_elev
 
 
-def _postprocess_coreg_apply_xarray_xarray(
+def _postprocess_coreg_apply_xarray(
     applied_elev: da.Array, out_transform: affine.Affine
 ) -> tuple[da.Array, affine.Affine]:
     """Post-processing and checks of apply for dask inputs."""
@@ -813,7 +813,7 @@ def _postprocess_coreg_apply(
             resampling=resampling,
         )
     elif isinstance(applied_elev, da.Array):
-        applied_elev, out_transform = _postprocess_coreg_apply_xarray_xarray(
+        applied_elev, out_transform = _postprocess_coreg_apply_xarray(
             applied_elev=applied_elev, out_transform=out_transform
         )