diff --git a/VAE/batch_slurm.sh b/VAE/batch_slurm.sh
index ff1474d9cb83f1c3bfcda5622939ce72ff5d1c9d..503885fe974e6ea14e753714bc241ceb6f516a8e 100755
--- a/VAE/batch_slurm.sh
+++ b/VAE/batch_slurm.sh
@@ -46,7 +46,8 @@ export FIDLE_OVERRIDE_VAE8_run_dir="./run/CelebA.$SLURM_JOB_ID"
 export FIDLE_OVERRIDE_VAE8_scale="1"
 export FIDLE_OVERRIDE_VAE8_image_size="(128,128)"
 export FIDLE_OVERRIDE_VAE8_enhanced_dir='{datasets_dir}/celeba/enhanced'
-export FIDLE_OVERRIDE_VAZ8_r_loss_factor="0.7"
+export FIDLE_OVERRIDE_VAE8_r_loss_factor="0.7"
+export FIDLE_OVERRIDE_VAE8_epochs="8"
 
 # ---- By default (no need to modify)
 #