diff --git a/mlpf/plotting/plot_validation.py b/mlpf/plotting/plot_validation.py
index c9607129a..f7e784c2b 100644
--- a/mlpf/plotting/plot_validation.py
+++ b/mlpf/plotting/plot_validation.py
@@ -851,8 +851,8 @@ def plot_purity_vs_kin(
         sample_label_coords[0], sample_label_coords[1], plot_sample_name, transform=ax.transAxes, fontsize=sample_label_fontsize, ha="left", va="top"
     )
     ax.text(
-        jet_label_coords_single[0],
-        jet_label_coords_single[1],
+        jet_label_coords[0],
+        jet_label_coords[1],
         jet_label_inclusive,
         transform=ax.transAxes,
         fontsize=addtext_fontsize,
@@ -891,8 +891,8 @@ def plot_purity_vs_kin(
         sample_label_coords[0], sample_label_coords[1], plot_sample_name, transform=ax.transAxes, fontsize=sample_label_fontsize, ha="left", va="top"
     )
     ax.text(
-        jet_label_coords_single[0],
-        jet_label_coords_single[1],
+        jet_label_coords[0],
+        jet_label_coords[1],
         jet_label_inclusive,
         transform=ax.transAxes,
         fontsize=addtext_fontsize,
@@ -942,8 +942,8 @@ def plot_purity_vs_kin(
         sample_label_coords[0], sample_label_coords[1], plot_sample_name, transform=ax.transAxes, fontsize=sample_label_fontsize, ha="left", va="top"
     )
     ax.text(
-        jet_label_coords_single[0],
-        jet_label_coords_single[1],
+        jet_label_coords[0],
+        jet_label_coords[1],
         jet_label_inclusive,
         transform=ax.transAxes,
         fontsize=addtext_fontsize,
diff --git a/scripts/joosep-desktop/cms-plots.sh b/scripts/joosep-desktop/cms-plots.sh
index cfc29edaf..ad80d568e 100755
--- a/scripts/joosep-desktop/cms-plots.sh
+++ b/scripts/joosep-desktop/cms-plots.sh
@@ -2,7 +2,7 @@
 
 export PYTHONPATH=`pwd`
 export QT_QPA_PLATFORM=offscreen
-IMG=~/HEP-KBFI/singularity/pytorch.simg\:2025-09-01
+IMG=~/HEP-KBFI/singularity/pytorch.simg\:2025-11-28
 NANO_PATH=/mnt/work/particleflow/CMSSW_15_0_5_mlpf_v2.6.0pre1_puppi_2372e2/cuda_False/
 
 # singularity exec -B /mnt/work/ $IMG  \