diff --git a/baselines/generate_figures.py b/baselines/generate_figures.py
index b23bf103..b99d8462 100644
--- a/baselines/generate_figures.py
+++ b/baselines/generate_figures.py
@@ -37,8 +37,8 @@
     'deepseg_2d': 'sct_deepseg_sc 2D',
     'deepseg_3d': 'sct_deepseg_sc 3D',
     'monai': 'contrast-agnostic',
-    'nnunet_2d': 'nnUNet 2D',
-    'nnunet_3d': 'nnUNet 3D'
+    'nnunet_2d': 'SCISeg 2D',
+    'nnunet_3d': 'SCISeg 3D'
     }
 
 METHODS_TO_LABEL_LESION = {