diff --git a/slurm/train_ResNet50.sh b/slurm/train_ResNet50.sh
index 74e703867dc8dbe4473721ba7b1479299abf569e..bc5f0a07b5df47a771fce34021c57b5d09b4c053 100644
--- a/slurm/train_ResNet50.sh
+++ b/slurm/train_ResNet50.sh
@@ -12,7 +12,7 @@
 module purge
 module load GCC/11.3.0  OpenMPI/4.1.4 TensorFlow/2.11.0-CUDA-11.7.0
 module load cuDNN/8.4.1.50-CUDA-11.7.0
-module load scikit-build/0.15.0
+module load scikit-learn/1.1.2
 
 # Run your script
 srun python ../Python/pokedex_ResNet50.py