diff --git a/test/test_training.py b/test/test_training.py
index 8e835f4e..d89a1597 100644
--- a/test/test_training.py
+++ b/test/test_training.py
@@ -172,7 +172,7 @@ def test_training(self):
         )
         self._run_inference_and_check_results(
             export_path, model_type, prediction_dir=prediction_dir,
-            inference_function=box_inference, expected_sa=0.95,
+            inference_function=box_inference, expected_sa=0.8,
         )
 
         # Check the model with interactive inference.
@@ -184,7 +184,7 @@ def test_training(self):
         )
         self._run_inference_and_check_results(
             export_path, model_type, prediction_dir=prediction_dir,
-            inference_function=iterative_inference, expected_sa=0.95,
+            inference_function=iterative_inference, expected_sa=0.8,
         )