diff --git a/tests/keras/test_keras_model_export.py b/tests/keras/test_keras_model_export.py
index 3fcd5367cf078..a8efc5898aeec 100644
--- a/tests/keras/test_keras_model_export.py
+++ b/tests/keras/test_keras_model_export.py
@@ -92,7 +92,7 @@ def test_model_save_load(model, model_path, data, predicted):
                                          pandas_df=pd.DataFrame(x),
                                          result_type="float")
     np.testing.assert_array_almost_equal(
-        np.array(spark_udf_preds), predicted.reshape(len(spark_udf_preds)), decimal=6)
+        np.array(spark_udf_preds), predicted.reshape(len(spark_udf_preds)), decimal=4)
 
 
 def test_model_log(tracking_uri_mock, model, data, predicted):  # pylint: disable=unused-argument