diff --git a/account_batch_payment_ux/models/account_batch_payment.py b/account_batch_payment_ux/models/account_batch_payment.py
index bd0532917..ccdc11747 100644
--- a/account_batch_payment_ux/models/account_batch_payment.py
+++ b/account_batch_payment_ux/models/account_batch_payment.py
@@ -41,3 +41,7 @@ def action_draft(self):
             self.payment_ids.move_id.is_move_sent = False
         self.payment_ids.unmark_as_sent()
         self.write({"state": "draft"})
+
+    def _valid_payment_states(self):
+        super()._valid_payment_states()
+        return ["in_process", "paid"]