diff --git a/src/aidsorb/data.py b/src/aidsorb/data.py
index 9d1f550..38d81c3 100644
--- a/src/aidsorb/data.py
+++ b/src/aidsorb/data.py
@@ -333,7 +333,7 @@ class PCDDataset(Dataset):
 
     .. note::
         * ``x`` and ``y`` are tensors of ``dtype=torch.float``.
-        * ``y`` has shape ``(len(labels),)``.
+        * ``y`` has shape ``(len(labels),)`` if ``transform_y=None``.
         * ``transform_x`` and ``transform_y`` expect :class:`~torch.Tensor` as
           input.