diff --git a/exir/backend/test/test_partitioner.py b/exir/backend/test/test_partitioner.py
index dedcfe52966..af1bd4d8aea 100644
--- a/exir/backend/test/test_partitioner.py
+++ b/exir/backend/test/test_partitioner.py
@@ -106,7 +106,7 @@ def partition(
 
         with self.assertRaisesRegex(
             AttributeError,
-            "can't set attribute 'spec'",
+            "(can't set attribute 'spec'|has no setter)",
         ):
             my_partitioner.spec = {"new_key": "new_value"}