diff --git a/src/axolotl/core/builders/causal.py b/src/axolotl/core/builders/causal.py
index d6d19607f..6ac647632 100644
--- a/src/axolotl/core/builders/causal.py
+++ b/src/axolotl/core/builders/causal.py
@@ -410,7 +410,7 @@ class HFCausalTrainerBuilder(TrainerBuilderBase):
 
     def build_collator(
         self,
-        training_args,  # type: "AxolotlTrainingArguments"
+        training_args,  # type: "AxolotlTrainingArguments"  # type: ignore
         is_eval=False,
         **kwargs,
     ):