diff --git a/membership_invoice_payment_mode/__manifest__.py b/membership_invoice_payment_mode/__manifest__.py
index 9217c82..1a512b1 100644
--- a/membership_invoice_payment_mode/__manifest__.py
+++ b/membership_invoice_payment_mode/__manifest__.py
@@ -3,7 +3,7 @@
 
 {
     "name": "membersip_invoice_payment_mode",
-    "version": "12.0.2.0.0",
+    "version": "16.0.1.0.0",
     "author": "Élabore",
     "maintainer": "False",
     "website": "elabore.coop",
diff --git a/membership_invoice_payment_mode/wizard/membership_invoice.py b/membership_invoice_payment_mode/wizard/membership_invoice.py
index 7bb2572..7b1e152 100644
--- a/membership_invoice_payment_mode/wizard/membership_invoice.py
+++ b/membership_invoice_payment_mode/wizard/membership_invoice.py
@@ -5,21 +5,24 @@ class MembershipInvoice(models.TransientModel):
     _inherit = "membership.invoice"
 
     payment_mode_id = fields.Many2one(
-        comodel_name='account.payment.mode', string="Payment Mode",
-        domain=[("payment_type", "=", 'inbound')],
-        )
+        comodel_name="account.payment.mode",
+        string="Payment Mode",
+        domain=[("payment_type", "=", "inbound")],
+    )
 
-    date_invoice = fields.Date(string='Invoice Date')
+    date_invoice = fields.Date(string="Invoice Date")
 
     def membership_invoice(self):
-        res = super(MembershipInvoice, self).membership_invoice()
+        res = super().membership_invoice()
         invoice_ids = None
-        for d in res['domain']:
-            if d[0] == 'id' and d[1] == 'in':
+        for d in res["domain"]:
+            if d[0] == "id" and d[1] == "in":
                 invoice_ids = d[2]
                 if invoice_ids:
-                    self.env['account.invoice'].browse(invoice_ids).write({
-                    'payment_mode_id':self.payment_mode_id.id,
-                    'date_invoice':self.date_invoice,
-                    })
+                    self.env["account.move"].browse(invoice_ids).write(
+                        {
+                            "payment_mode_id": self.payment_mode_id.id,
+                            "invoice_date": self.date_invoice,
+                        },
+                    )
         return res