diff --git a/atlas_server/src/static/training/predict.job.sh b/atlas_server/src/static/training/predict.job.sh
index b2d890a891e1a68397b6fc44f8c9bd0865b4cd65..54ca93130b6b34cbc493c02cf31ae03fcbb155b4 100644
--- a/atlas_server/src/static/training/predict.job.sh
+++ b/atlas_server/src/static/training/predict.job.sh
@@ -5,13 +5,10 @@ module purge
 
 export SOFTWARE_STAGE=2022
 export OTHERSTAGES=/p/software/jurecadc/otherstages
-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/
+source ${HOME}/galaxy/init.sh
 export ATLAS_MODULE=atlas
 
-module use $OTHERSTAGES &> /dev/null
+module use ${OTHERSTAGES} &> /dev/null
 module use ${SOFTWARE_STAGE_MODULES} &> /dev/null
 
 module load ${ATLAS_MODULE}
diff --git a/atlas_server/src/static/training/train.job.sh b/atlas_server/src/static/training/train.job.sh
index 5a504dac3cd0c84564276c71de64e400a37adcba..146f415dbe0d1eb16ddadb5ba595e947f4b93888 100644
--- a/atlas_server/src/static/training/train.job.sh
+++ b/atlas_server/src/static/training/train.job.sh
@@ -5,13 +5,10 @@ module purge
 
 export SOFTWARE_STAGE=2022
 export OTHERSTAGES=/p/software/jurecadc/otherstages
-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/
+source ${HOME}/galaxy/init.sh
 export ATLAS_MODULE=atlas
 
-module use $OTHERSTAGES &> /dev/null
+module use ${OTHERSTAGES} &> /dev/null
 module use ${SOFTWARE_STAGE_MODULES} &> /dev/null
 
 module load ${ATLAS_MODULE}