diff --git a/account_payment_partner/migrations/9.0.1.0.0/post-migration.py b/account_payment_partner/migrations/9.0.1.0.0/post-migration.py
new file mode 100644
index 000000000..09c906855
--- /dev/null
+++ b/account_payment_partner/migrations/9.0.1.0.0/post-migration.py
@@ -0,0 +1,19 @@
+# -*- coding: utf-8 -*-
+# © 2016 Sergio Teruel <sergio.teruel@tecnativa.com>
+# License AGPL-3 - See http://www.gnu.org/licenses/agpl-3.0.html
+
+from openupgradelib import openupgrade
+
+
+@openupgrade.migrate(use_env=True)
+def migrate(env, version):
+    # Copy mandate_id to account_move_line
+    sql = """
+    UPDATE account_move_line aml
+    SET payment_mode_id = ai.payment_mode_id
+    FROM account_invoice ai
+    WHERE ai.id=aml.invoice_id
+    AND aml.invoice_id IS NOT NULL
+    AND ai.payment_mode_id IS NOT NULL;
+    """
+    openupgrade.logged_query(env.cr, sql)