diff --git a/training/scripts/train.py b/training/scripts/train.py
index bde830a..fd56ccb 100755
--- a/training/scripts/train.py
+++ b/training/scripts/train.py
@@ -99,7 +99,7 @@ def train(config_path):
     
     trainer = SFTTrainer(
         model=model,
-        tokenizer=tokenizer,
+        processing_class=tokenizer,
         train_dataset=dataset["train"],
         eval_dataset=dataset["test"],
         args=training_args,