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 b885f640d941..bd508a4d7b1a 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
@@ -319,7 +319,7 @@ trait ReadXlmRoBertaForQuestionAnsweringDLModel
     instance.getEngine match {
       case TensorFlow.name =>
         val tfWrapper =
-          readTensorflowModel(path, spark, "xlm_roberta_classification_tf", initAllTables = false)
+          readTensorflowModel(path, spark, "xlm_roberta_classification_tensorflow", initAllTables = false)
         instance.setModelIfNotSet(spark, Some(tfWrapper), None, spp)
       case ONNX.name =>
         val onnxWrapper =
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 366b0ce0fa8a..782cbd01ed5b 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
@@ -370,7 +370,7 @@ trait ReadXlmRoBertaForSequenceDLModel
     instance.getEngine match {
       case TensorFlow.name =>
         val tfWrapper =
-          readTensorflowModel(path, spark, "xlm_roberta_classification_tf", initAllTables = false)
+          readTensorflowModel(path, spark, "xlm_roberta_classification_tensorflow", initAllTables = false)
         instance.setModelIfNotSet(spark, Some(tfWrapper), None, spp)
       case ONNX.name =>
         val onnxWrapper =
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 f9f933c1f8d0..2f6cbed1db5f 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
@@ -342,7 +342,7 @@ trait ReadXlmRoBertaForTokenDLModel
     instance.getEngine match {
       case TensorFlow.name =>
         val tfWrapper =
-          readTensorflowModel(path, spark, "xlm_roberta_classification_tf", initAllTables = false)
+          readTensorflowModel(path, spark, "xlm_roberta_classification_tensorflow", initAllTables = false)
         instance.setModelIfNotSet(spark, Some(tfWrapper), None, spp)
       case ONNX.name =>
         val onnxWrapper =