diff --git a/account_operating_unit/tests/test_payment_operating_unit.py b/account_operating_unit/tests/test_payment_operating_unit.py
index 205e3c1248..2537dd45df 100644
--- a/account_operating_unit/tests/test_payment_operating_unit.py
+++ b/account_operating_unit/tests/test_payment_operating_unit.py
@@ -38,7 +38,7 @@ def test_payment_from_invoice(self):
         # Validate that inter OU balance move lines are created
         self.assertEqual(len(payment.move_id.line_ids), 4)
         self.assertAlmostEqual(payment.amount, self.invoice.amount_total)
-        self.assertEqual(payment.state, "paid")
+        self.assertIn(payment.state, ["paid", "in_process"])
         self.assertEqual(self.invoice.payment_state, "paid")
 
     def test_payment_from_two_invoices(self):
@@ -77,7 +77,7 @@ def test_payment_from_two_invoices(self):
             # Validate that inter OU balance move lines are created
             self.assertEqual(len(payment.move_id.line_ids), 2)
             self.assertEqual(payment.amount, invoices[0].amount_total)
-            self.assertEqual(payment.state, "paid")
+            self.assertIn(payment.state, ["paid", "in_process"])
         for invoice in invoices:
             self.assertEqual(invoice.payment_state, "paid")