diff --git a/src/main/scala/com/johnsnowlabs/nlp/annotators/classifier/dl/XlmRoBertaForQuestionAnswering.scala b/src/main/scala/com/johnsnowlabs/nlp/annotators/classifier/dl/XlmRoBertaForQuestionAnswering.scala
index 3cb512d56195b9..b885f640d94145 100644
--- a/src/main/scala/com/johnsnowlabs/nlp/annotators/classifier/dl/XlmRoBertaForQuestionAnswering.scala
+++ b/src/main/scala/com/johnsnowlabs/nlp/annotators/classifier/dl/XlmRoBertaForQuestionAnswering.scala
@@ -306,7 +306,7 @@ trait ReadXlmRoBertaForQuestionAnsweringDLModel
     with ReadSentencePieceModel {
   this: ParamsAndFeaturesReadable[XlmRoBertaForQuestionAnswering] =>
 
-  override val tfFile: String = "xlm_roberta_classification_tf"
+  override val tfFile: String = "xlm_roberta_classification_tensorflow"
   override val onnxFile: String = "xlm_roberta_classification_onnx"
   override val sppFile: String = "xlmroberta_spp"
 
diff --git a/src/main/scala/com/johnsnowlabs/nlp/annotators/classifier/dl/XlmRoBertaForSequenceClassification.scala b/src/main/scala/com/johnsnowlabs/nlp/annotators/classifier/dl/XlmRoBertaForSequenceClassification.scala
index 8f9f12c5a2af8f..366b0ce0fa8ad6 100644
--- a/src/main/scala/com/johnsnowlabs/nlp/annotators/classifier/dl/XlmRoBertaForSequenceClassification.scala
+++ b/src/main/scala/com/johnsnowlabs/nlp/annotators/classifier/dl/XlmRoBertaForSequenceClassification.scala
@@ -357,7 +357,7 @@ trait ReadXlmRoBertaForSequenceDLModel
     with ReadSentencePieceModel {
   this: ParamsAndFeaturesReadable[XlmRoBertaForSequenceClassification] =>
 
-  override val tfFile: String = "xlm_roberta_classification_tf"
+  override val tfFile: String = "xlm_roberta_classification_tensorflow"
   override val onnxFile: String = "xlm_roberta_classification_onnx"
   override val sppFile: String = "xlmroberta_spp"
 
diff --git a/src/main/scala/com/johnsnowlabs/nlp/annotators/classifier/dl/XlmRoBertaForTokenClassification.scala b/src/main/scala/com/johnsnowlabs/nlp/annotators/classifier/dl/XlmRoBertaForTokenClassification.scala
index 2f5328583a2712..f9f933c1f8d018 100644
--- a/src/main/scala/com/johnsnowlabs/nlp/annotators/classifier/dl/XlmRoBertaForTokenClassification.scala
+++ b/src/main/scala/com/johnsnowlabs/nlp/annotators/classifier/dl/XlmRoBertaForTokenClassification.scala
@@ -328,7 +328,7 @@ trait ReadXlmRoBertaForTokenDLModel
     with ReadSentencePieceModel {
   this: ParamsAndFeaturesReadable[XlmRoBertaForTokenClassification] =>
 
-  override val tfFile: String = "xlm_roberta_classification_tf"
+  override val tfFile: String = "xlm_roberta_classification_tensorflow"
   override val onnxFile: String = "xlm_roberta_classification_onnx"
   override val sppFile: String = "xlmroberta_spp"