diff --git a/mace_mp_0b/mace-agnesi-small.sh b/mace_mp_0b/mace-agnesi-small.sh
index 17aaa47..8fbbfa1 100644
--- a/mace_mp_0b/mace-agnesi-small.sh
+++ b/mace_mp_0b/mace-agnesi-small.sh
@@ -1,8 +1,8 @@
 srun python mace/mace/cli/run_train.py \
     --name="MACE_large_agnesi" \
-    --train_file=".../mace_large_hp/mace_mp_data/train" \
-    --valid_file=".../mace_large_hp/mace_mp_data/valid" \
-    --statistics_file=".../mace_mp_data/mptrj-gga-ggapu-statistics.json" \
+    --train_file=".../train" \
+    --valid_file=".../valid" \
+    --statistics_file=".../mptrj-gga-ggapu-statistics.json" \
     --loss='universal' \
     --energy_weight=1 \
     --forces_weight=10 \