diff --git a/qim3d/io/_ome_zarr.py b/qim3d/io/_ome_zarr.py
index 0cf4803e4fa02888fc9a00e0582048e4f62abea3..983aab0174de9efd110b1d8eff5010f3bb2e7d54 100644
--- a/qim3d/io/_ome_zarr.py
+++ b/qim3d/io/_ome_zarr.py
@@ -70,10 +70,10 @@ class OMEScaler(
         This method performs multi-scale downsampling on a 3D dataset, generating image pyramids. It processes the data in chunks using Dask.
 
         Args:
-            base (dask.array): The 3D array (volume) to be downsampled. Must be a Dask array for chunked processing.
+            base (dask.array.core.array): The 3D array (volume) to be downsampled. Must be a Dask array for chunked processing.
 
         Returns:
-            list of dask.array: A list of downsampled volumes, where each element represents a different scale. The first element corresponds to the original resolution, and subsequent elements represent progressively downsampled versions.
+            list of dask.array.core.Array: A list of downsampled volumes, where each element represents a different scale. The first element corresponds to the original resolution, and subsequent elements represent progressively downsampled versions.
 
         The downsampling process occurs scale by scale, using the following steps:
         - For each scale, the array is resized based on the downscale factor, computed as a function of the current scale level.
diff --git a/qim3d/io/_saving.py b/qim3d/io/_saving.py
index 830674a2d3d83f442c1b4772cdb840f3da88c345..ff713a4f7afd7a3d67de3e01823b717695e9e699 100644
--- a/qim3d/io/_saving.py
+++ b/qim3d/io/_saving.py
@@ -260,7 +260,7 @@ class DataSaver:
 
         Args:
             path (str): The path to the Zarr array on disk.
-            data (dask.array): The Dask array to be saved to disk.
+            data (dask.array.core.Array): The Dask array to be saved to disk.
 
         Returns:
             zarr.core.Array: The Zarr array saved on disk.