diff --git a/src/mednet/libs/segmentation/engine/dumper.py b/src/mednet/libs/segmentation/engine/dumper.py
index a4822e1c2a751f0c28bb482e26630ac95b9ebac0..742af03eb1a06d1ae8ef4064de7e705e692cf0e6 100644
--- a/src/mednet/libs/segmentation/engine/dumper.py
+++ b/src/mednet/libs/segmentation/engine/dumper.py
@@ -66,7 +66,7 @@ def run(
             directory where the HDF5 file was saved.
         """
         name = sample[1]["name"][0]
-        target = sample[1]["target"][0]
+        target = sample[0]["target"][0]
         stem = pathlib.Path(name).with_suffix(".hdf5")
         dest = output_folder / stem
         tqdm.tqdm.write(f"`{name}` -> `{str(dest)}`")