diff --git a/trlx/trainer/accelerate_base_trainer.py b/trlx/trainer/accelerate_base_trainer.py
index cf3b58c5e..d9b7ae8fc 100644
--- a/trlx/trainer/accelerate_base_trainer.py
+++ b/trlx/trainer/accelerate_base_trainer.py
@@ -431,7 +431,7 @@ def evaluate(self):  # noqa: C901
                         samples=str_samples,
                         prompts=str_prompts,
                         outputs=str_outputs,
-                        model_tok=self.tokenizer,
+                        tokenizer=self.tokenizer,
                         **metadata,
                     )
                     if isinstance(rewards[0], torch.Tensor):