diff --git a/atlas_server/src/static/training/predict.job.sh b/atlas_server/src/static/training/predict.job.sh
index 053f15b0f1098c65373eeaf96717649a933781e9..82a4658106494a533586d01dbde0ea8cd7bd7a6b 100644
--- a/atlas_server/src/static/training/predict.job.sh
+++ b/atlas_server/src/static/training/predict.job.sh
@@ -20,7 +20,7 @@ export OUTPUT_SPACING=16
 
 # Set annotation directory (assume we are in the experiment folder)
 export ATLAS_ANNOTATION_DIR=$(pwd)/annotations/
-export ATLAS_ANNOTATION_FILE="B${BRAIN}_{slice:04d}_annotations.hdf5"
+export ATLAS_ANNOTATION_FILE="B{brain:02d}_{slice:04d}_annotations.hdf5"
 
 # Make prediction and run evaluation
 srun --cpus-per-task 4 atlas dl predict $(pwd) --verbose --output-spacing ${OUTPUT_SPACING} --overwrite --save-argmax
diff --git a/atlas_server/src/static/training/train.job.sh b/atlas_server/src/static/training/train.job.sh
index f8dea8fa5815c1110a421ade9f2a1373bde8791b..3b53924a7b472431dda618efa0256b88a4e294d1 100644
--- a/atlas_server/src/static/training/train.job.sh
+++ b/atlas_server/src/static/training/train.job.sh
@@ -19,7 +19,7 @@ export SCRATCH_USER=/p/scratch/cjinm16/schiffer1/
 
 # Set annotation directory (assume we are in the experiment folder)
 export ATLAS_ANNOTATION_DIR=$(pwd)/annotations/
-export ATLAS_ANNOTATION_FILE="B${BRAIN}_{slice:04d}_annotations.hdf5"
+export ATLAS_ANNOTATION_FILE="B{brain:02d}_{slice:04d}_annotations.hdf5"
 
 # Some more variables
 INPUT_SPACING=1
diff --git a/atlas_server/src/static/training/train_predict.job.sh b/atlas_server/src/static/training/train_predict.job.sh
index 6979d58e22d908d7cadb285360e2b1765ac683e4..116f78431cae3dba3baf5b0f53155774c7916cf6 100644
--- a/atlas_server/src/static/training/train_predict.job.sh
+++ b/atlas_server/src/static/training/train_predict.job.sh
@@ -19,7 +19,7 @@ export SCRATCH_USER=/p/scratch/cjinm16/schiffer1/
 
 # Set annotation directory (assume we are in the experiment folder)
 export ATLAS_ANNOTATION_DIR=$(pwd)/annotations/
-export ATLAS_ANNOTATION_FILE="B${BRAIN}_{slice:04d}_annotations.hdf5"
+export ATLAS_ANNOTATION_FILE="B{brain:02d}_{slice:04d}_annotations.hdf5"
 
 # Some more variables
 INPUT_SPACING=1