diff --git a/tests/test_rlearner.py b/tests/test_rlearner.py
index 1e009f9..a5dc990 100644
--- a/tests/test_rlearner.py
+++ b/tests/test_rlearner.py
@@ -138,4 +138,4 @@ def test_rlearner_onnx(treatment_model_factory, onnx_converter, is_classificatio
         ["tau"],
         {"input": onnx_X},
     )
-    np.testing.assert_allclose(ml.predict(X, True, "overall"), pred_onnx[0], atol=1e-5)
+    np.testing.assert_allclose(ml.predict(X, True, "overall"), pred_onnx[0], atol=5e-4)
diff --git a/tests/test_tlearner.py b/tests/test_tlearner.py
index ccfeb62..2bfd8c2 100644
--- a/tests/test_tlearner.py
+++ b/tests/test_tlearner.py
@@ -154,4 +154,4 @@ def test_tlearner_onnx(nuisance_model_factory, onnx_converter, is_classification
         ["tau"],
         {"input": onnx_X},
     )
-    np.testing.assert_allclose(ml.predict(X, True, "overall"), pred_onnx[0], atol=1e-4)
+    np.testing.assert_allclose(ml.predict(X, True, "overall"), pred_onnx[0], atol=5e-4)
diff --git a/tests/test_xlearner.py b/tests/test_xlearner.py
index be6c110..521e06b 100644
--- a/tests/test_xlearner.py
+++ b/tests/test_xlearner.py
@@ -142,4 +142,4 @@ def test_xlearner_onnx(
         ["tau", "Div_1_C"],
         {"input": X.astype(np.float32)},
     )
-    np.testing.assert_allclose(ml.predict(X, True, "overall"), pred_onnx[0], atol=1e-5)
+    np.testing.assert_allclose(ml.predict(X, True, "overall"), pred_onnx[0], atol=5e-4)