diff --git a/monai/transforms/post/array.py b/monai/transforms/post/array.py
index 42d38c4b24..7e687e6203 100644
--- a/monai/transforms/post/array.py
+++ b/monai/transforms/post/array.py
@@ -960,7 +960,7 @@ def __call__(self, img: NdarrayOrTensor) -> NdarrayOrTensor:
         """
         Args:
             img: Input image on which the distance transform shall be run.
-                For CuPy shape must be ([channel, ]spatial_dim1, spatial_dim2 [, spatial_dim3]), so 2D, 3D and 4D input only.
+                channel first array, must have shape: (num_channels, H[, W, ..., ])
                 If you need to run the transform on other shapes, use the ``force_scipy`` flag.
                 4D input gets passed channel-wise to cupy.