diff --git a/src/extraction/core/functions/distributors.py b/src/extraction/core/functions/distributors.py
index 8b35c86f6adb04fa1912ab5780605d1138b4109e..92152a88408c83ab6443d450223fe418819bbb83 100644
--- a/src/extraction/core/functions/distributors.py
+++ b/src/extraction/core/functions/distributors.py
@@ -13,7 +13,7 @@ def trap_apply(cell_fun, cell_masks, *args, **kwargs):
     cell_fun: function
         Function to apply to the cell (from extraction/cell.py)
     cell_masks: 3d array
-        Segmentation masks for the cells. Note that cells are the last dimension N (Y,X,N)
+        Segmentation masks for the cells. Note that cells are in the first dimension (N, Y,X)
     *args: tuple
         Trap_image and any other arguments to pass if needed to custom functions.
     **kwargs: dict