diff --git a/results/kr-vs-kp/stage4_fix/losses.png b/results/kr-vs-kp/stage4_fix/losses.png
index 08d7af6f6354d596f53a73afd924a72225506ee7..64d6edffd85118de960d2303d212a7ebcd0a94b9 100644
Binary files a/results/kr-vs-kp/stage4_fix/losses.png and b/results/kr-vs-kp/stage4_fix/losses.png differ
diff --git a/results/steel-plates/stage4_fix/losses.png b/results/steel-plates/stage4_fix/losses.png
index f473402fc01d0f6d2d8c86941bae6862beb4fb04..d18c50afcb8b6c29819c576588282144f3a077d6 100644
Binary files a/results/steel-plates/stage4_fix/losses.png and b/results/steel-plates/stage4_fix/losses.png differ
diff --git a/scripts/run_compute_results.sh b/scripts/run_compute_results.sh
index d039053a1b7bda6ceeab95ff3ae9509dc5a54e0c..1e8cf514ad361f31cbcf204696a766503eed1782 100644
--- a/scripts/run_compute_results.sh
+++ b/scripts/run_compute_results.sh
@@ -1,4 +1,4 @@
-for dataset in breast_cancer #kr-vs-kp steel-plates
+for dataset in breast_cancer kr-vs-kp steel-plates
 do
     python code/compute_results.py --stage=4 --experiment_ids 1 2 3 --dataset_name=$dataset --models_dir=models/$dataset/stage5
 done