diff --git a/atlas_server/src/static/training/predict.job.sh b/atlas_server/src/static/training/predict.job.sh
index 49cb24edcd8f0e2c5ff36f5fba71b1d827bc97ee..b2d890a891e1a68397b6fc44f8c9bd0865b4cd65 100644
--- a/atlas_server/src/static/training/predict.job.sh
+++ b/atlas_server/src/static/training/predict.job.sh
@@ -9,7 +9,7 @@ export SOFTWARE_DIR=/p/project/cjinm16/schiffer1/software
 export SOFTWARE_STAGES_DIR=${SOFTWARE_DIR}/stages/
 export SOFTWARE_STAGE_DIR=${SOFTWARE_STAGES_DIR}/${SOFTWARE_STAGE}
 export SOFTWARE_STAGE_MODULES=${SOFTWARE_STAGE_DIR}/modules/
-export ATLAS_MODULE=atlas/OpenMPI_PyTorch
+export ATLAS_MODULE=atlas
 
 module use $OTHERSTAGES &> /dev/null
 module use ${SOFTWARE_STAGE_MODULES} &> /dev/null
diff --git a/atlas_server/src/static/training/train.job.sh b/atlas_server/src/static/training/train.job.sh
index 92cc8eefd416dd30c08e19de2c83451579f60007..5a504dac3cd0c84564276c71de64e400a37adcba 100644
--- a/atlas_server/src/static/training/train.job.sh
+++ b/atlas_server/src/static/training/train.job.sh
@@ -9,7 +9,7 @@ export SOFTWARE_DIR=/p/project/cjinm16/schiffer1/software
 export SOFTWARE_STAGES_DIR=${SOFTWARE_DIR}/stages/
 export SOFTWARE_STAGE_DIR=${SOFTWARE_STAGES_DIR}/${SOFTWARE_STAGE}
 export SOFTWARE_STAGE_MODULES=${SOFTWARE_STAGE_DIR}/modules/
-export ATLAS_MODULE=atlas/OpenMPI_PyTorch
+export ATLAS_MODULE=atlas
 
 module use $OTHERSTAGES &> /dev/null
 module use ${SOFTWARE_STAGE_MODULES} &> /dev/null