diff --git a/qim3d/io/_ome_zarr.py b/qim3d/io/_ome_zarr.py
index c6f1327e23bb0812af88d8645276bd5b0951cc04..47d0b42d4d8b5647fc7b5747504de65b3d8f06fd 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 d173ed7592800bb651367ba5e4e39ab74ff1299e..6186b1e63b8e95f7d4e8c9c52b22be3b54c60429 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.