diff --git a/src/mednet/libs/segmentation/engine/evaluator.py b/src/mednet/libs/segmentation/engine/evaluator.py
index 019171043df6577cb75463bc98126838ebccd761..6da6282c6758ddb4948a5f622839c343100afc0d 100644
--- a/src/mednet/libs/segmentation/engine/evaluator.py
+++ b/src/mednet/libs/segmentation/engine/evaluator.py
@@ -364,7 +364,7 @@ def _evaluate_sample_worker(
     retval = _sample_measures(prediction, target, mask, steps)
 
     if output_folder is not None:
-        fullpath = output_folder / name / f"{stem}.csv"
+        fullpath = output_folder / name / f"{pathlib.Path(stem).stem}.csv"
         tqdm.write(f"Saving {fullpath}...")
         fullpath.parent.mkdir(parents=True, exist_ok=True)
         retval.to_csv(fullpath)
diff --git a/src/mednet/libs/segmentation/scripts/predict.py b/src/mednet/libs/segmentation/scripts/predict.py
index e96d25a3b905967b5497fd15df2773366b2f08e3..b91400ac93dc98e43ce897c4efcc7e2f2c47bc93 100644
--- a/src/mednet/libs/segmentation/scripts/predict.py
+++ b/src/mednet/libs/segmentation/scripts/predict.py
@@ -115,7 +115,7 @@ def predict(
     for split_name, split in predictions.items():
         pred_paths = []
         for sample in split:
-            hdf5_path = output_folder / f"{sample[0]}.hdf5"
+            hdf5_path = output_folder / f"{pathlib.Path(sample[0]).stem}.hdf5"
             _save_hdf5(sample[3], sample[1], sample[2], hdf5_path)
             pred_paths.append([str(sample[0]), str(hdf5_path)])
         json_predictions[split_name] = pred_paths