diff --git a/_unittests/ut_helpers/test_torch_helper.py b/_unittests/ut_helpers/test_torch_helper.py
index 1070672c..e5fc67d8 100644
--- a/_unittests/ut_helpers/test_torch_helper.py
+++ b/_unittests/ut_helpers/test_torch_helper.py
@@ -240,6 +240,7 @@ def test_replace_string_by_dynamic(self):
                 "_DimHint(type=<_DimHintType.DYNAMIC:3>,min=None,max=None,_factory=True)",
                 "DYN",
             )
+            .replace("DimHint(DYNAMIC)", "DYN")
         )
         self.assertEqual(
             "{'input_ids':{0:DYN,1:DYN},'attention_mask':({0:DYN,1:DYN},),'position_ids':[{0:DYN,1:DYN}]}",
diff --git a/requirements.txt b/requirements.txt
index 20f02745..0e11c176 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,3 +1,4 @@
+ml_dtypes
 numpy
 onnx>=1.16.0
 onnxruntime>=1.23