diff --git a/GANDLF/data/inference_dataloader_histopath.py b/GANDLF/data/inference_dataloader_histopath.py
index 97508df55..3e7093c1b 100644
--- a/GANDLF/data/inference_dataloader_histopath.py
+++ b/GANDLF/data/inference_dataloader_histopath.py
@@ -139,7 +139,7 @@ def __getitem__(self, idx):
 
         # this is to ensure that channels come at the beginning
         # patch = patch.transpose([2, 0, 1])
-        patch = patch.transpose(0)
+        patch = patch.transpose(0) # check the documentation
         # this is to ensure that we always have a z-stack before applying any torchio transforms
         patch = np.expand_dims(patch, axis=-1)
         if self.transform is not None: