diff --git a/benchmarks/calo_pid/calo_pid.org b/benchmarks/calo_pid/calo_pid.org
index 965af1400e13dcc7229908bda4b2a0a7aa251a9a..c9ec2fa92d8a0fff53235a8936cbf0c7eb15d851 100644
--- a/benchmarks/calo_pid/calo_pid.org
+++ b/benchmarks/calo_pid/calo_pid.org
@@ -363,6 +363,7 @@ if "_EcalEndcapNParticleIDOutput_probability_tensor_floatData" in pi_train.field
     eval_proba = ak.concatenate([pi_eval_proba, e_eval_proba])
 
     plt.hist(clf.predict_proba(eval_x.to_numpy())[:,1] - eval_proba[:,1].to_numpy())
+    plt.savefig(output_dir / f"proba_diff.pdf", bbox_inches="tight")
     plt.show()
 else:
     print("EcalEndcapNParticleIDOutput not present")