diff --git a/src/sagemaker/djl_inference/model.py b/src/sagemaker/djl_inference/model.py
index 8c724a6502..2a23cba9ec 100644
--- a/src/sagemaker/djl_inference/model.py
+++ b/src/sagemaker/djl_inference/model.py
@@ -148,7 +148,7 @@ def _infer_engine(self) -> Optional[str]:
             return self.engine
 
         if self.task == "text-embedding":
-            return "OnnxRuntime"
+            return "Rust"
         return "Python"
 
     def _infer_image_uri(self):