diff --git a/prompt_tuner.py b/prompt_tuner.py
index f172c6a1..2fbd6ee2 100644
--- a/prompt_tuner.py
+++ b/prompt_tuner.py
@@ -251,7 +251,7 @@ class TrainerBase(abc.ABC):
         return True
 
     def get_tokenizer(self) -> transformers.PreTrainedTokenizerBase:
-        return get_tokenizer(self.ckpt_path)
+        return get_tokenizer(self.data.ckpt_path)
     
     def save_data(self):
         pass