diff --git a/account_invoice_update_wizard/wizard/account_move_update.py b/account_invoice_update_wizard/wizard/account_move_update.py
index b95a989..b25edef 100644
--- a/account_invoice_update_wizard/wizard/account_move_update.py
+++ b/account_invoice_update_wizard/wizard/account_move_update.py
@@ -14,6 +14,7 @@ class AccountMoveUpdate(models.TransientModel):
     invoice_id = fields.Many2one(
         'account.move', string='Invoice', required=True,
         readonly=True)
+    bank_partner_id = fields.Many2one(related="invoice_id.bank_partner_id")
     move_type = fields.Selection(related='invoice_id.move_type')
     company_id = fields.Many2one(related='invoice_id.company_id')
     partner_id = fields.Many2one(related='invoice_id.partner_id')
@@ -56,17 +57,6 @@ class AccountMoveUpdate(models.TransientModel):
             }])
         return res
 
-    @api.onchange('move_type')
-    def move_type_on_change(self):
-        res = {'domain': {}}
-        if self.move_type in ('out_invoice', 'out_refund'):
-            res['domain']['partner_bank_id'] =\
-                "[('partner_id.ref_company_ids', 'in', [company_id])]"
-        else:
-            res['domain']['partner_bank_id'] =\
-                "[('partner_id', '=', partner_id)]"
-        return res
-
     def _prepare_invoice(self):
         vals = {}
         inv = self.invoice_id
diff --git a/account_invoice_update_wizard/wizard/account_move_update_view.xml b/account_invoice_update_wizard/wizard/account_move_update_view.xml
index df51010..ff2cd62 100644
--- a/account_invoice_update_wizard/wizard/account_move_update_view.xml
+++ b/account_invoice_update_wizard/wizard/account_move_update_view.xml
@@ -22,7 +22,7 @@
                     <field name="invoice_origin" invisible="move_type == 'entry'"/>
                     <!-- update of payment term is broken -->
                     <!-- <field name="invoice_payment_term_id" widget="selection"/>-->
-                    <field name="partner_bank_id" invisible="move_type == 'entry'"/>
+                    <field name="partner_bank_id" invisible="move_type == 'entry'" domain="[('partner_id', '=', bank_partner_id)]"/>
                     <field name="invoice_user_id" options="{'no_open': True, 'no_create': True, 'no_create_edit': True}" invisible="move_type == 'entry'"/>
                 </group>
                 <group name="lines">