diff --git a/hyperlora/intx_sft.py b/hyperlora/intx_sft.py
index 665258d..c34926e 100644
--- a/hyperlora/intx_sft.py
+++ b/hyperlora/intx_sft.py
@@ -391,7 +391,7 @@ def main(output_dir):
     if "validation" in tokenized_ds:
         val_ds["validation"] = tokenized_ds["validation"]
         val_ds_size = len(val_ds["validation"])
-        val_indices = np.random.permutation(val_ds_size)[: args.max_val_samples]
+        val_indices = np.random.permutation(val_ds_size)[: data_args.max_val_samples]
         val_ds["validation"] = val_ds["validation"].select(val_indices)
     test_ds = tokenized_ds.get("test", None)