diff --git a/test/asynchronous/test_encryption.py b/test/asynchronous/test_encryption.py
index 337dba0f64..4517c9f917 100644
--- a/test/asynchronous/test_encryption.py
+++ b/test/asynchronous/test_encryption.py
@@ -1268,7 +1268,7 @@ async def test_06_insert_fails_over_16MiB(self):
         with self.assertRaises(BulkWriteError) as ctx:
             await self.coll_encrypted.bulk_write([InsertOne(doc)])
         err = ctx.exception.details["writeErrors"][0]
-        self.assertEqual(2, err["code"])
+        self.assertIn(err["code"], [2, 10334])
         self.assertIn("object to insert too large", err["errmsg"])
 
 
diff --git a/test/test_encryption.py b/test/test_encryption.py
index 46d8c785c4..fb18cc2dd0 100644
--- a/test/test_encryption.py
+++ b/test/test_encryption.py
@@ -1264,7 +1264,7 @@ def test_06_insert_fails_over_16MiB(self):
         with self.assertRaises(BulkWriteError) as ctx:
             self.coll_encrypted.bulk_write([InsertOne(doc)])
         err = ctx.exception.details["writeErrors"][0]
-        self.assertEqual(2, err["code"])
+        self.assertIn(err["code"], [2, 10334])
         self.assertIn("object to insert too large", err["errmsg"])