diff --git a/scripts/hpc/train.pbs b/scripts/hpc/train.pbs
index edb1b7e..8556f2d 100644
--- a/scripts/hpc/train.pbs
+++ b/scripts/hpc/train.pbs
@@ -36,7 +36,8 @@ export MUJOCO_GL=egl
 export WANDB_DIR="$SCRATCH_RUNDIR"
 
 python src/train.py \
-    --config configs/hpc/smoke_test.yaml
+    --config configs/hpc/smoke_test.yaml \
+    --run_dir "$SCRATCH_RUNDIR"
     # ^^^ Replace with your production config, e.g. configs/production_training.yaml
 
 cp -r "$SCRATCH_RUNDIR/." "$DATA_RUNDIR/"