diff --git a/tests/models/vitmatte/test_image_processing_vitmatte.py b/tests/models/vitmatte/test_image_processing_vitmatte.py
index 638bcfb958ae89..288ed53d190d00 100644
--- a/tests/models/vitmatte/test_image_processing_vitmatte.py
+++ b/tests/models/vitmatte/test_image_processing_vitmatte.py
@@ -213,5 +213,6 @@ def test_image_processor_preprocess_arguments(self):
                 warnings.simplefilter("always")
                 image_processor(image, trimaps=trimap, extra_argument=True)
 
-            self.assertEqual(len(raised_warnings), 1)
-            self.assertIn("extra_argument", str(raised_warnings[0].message))
+            messages = " ".join([str(w.message) for w in raised_warnings])
+            self.assertGreaterEqual(len(raised_warnings), 1)
+            self.assertIn("extra_argument", messages)
diff --git a/tests/test_image_processing_common.py b/tests/test_image_processing_common.py
index fa267cedf8d3b5..a3fbb6eeb78178 100644
--- a/tests/test_image_processing_common.py
+++ b/tests/test_image_processing_common.py
@@ -454,8 +454,9 @@ def test_image_processor_preprocess_arguments(self):
                     warnings.simplefilter("always")
                     image_processor(inputs, extra_argument=True)
 
-                self.assertEqual(len(raised_warnings), 1)
-                self.assertIn("extra_argument", str(raised_warnings[0].message))
+                messages = " ".join([str(w.message) for w in raised_warnings])
+                self.assertGreaterEqual(len(raised_warnings), 1)
+                self.assertIn("extra_argument", messages)
                 is_tested = True
 
         if not is_tested: