diff --git a/xdem/coreg/workflows.py b/xdem/coreg/workflows.py
index 37ecb98e..04999144 100644
--- a/xdem/coreg/workflows.py
+++ b/xdem/coreg/workflows.py
@@ -161,7 +161,7 @@ def dem_coregistration(
     random_state: int | np.random.Generator | None = None,
     plot: bool = False,
     out_fig: str = None,
-    estimated_initial_shift: list[float] | tuple[float, float] = None,
+    estimated_initial_shift: list[Number] | tuple[Number, Number] | None = None,
 ) -> tuple[DEM, Coreg | CoregPipeline, pd.DataFrame, NDArrayf]:
     """
     A one-line function to coregister a selected DEM to a reference DEM.