diff --git a/account_edi_simple_pdf/models/account_move.py b/account_edi_simple_pdf/models/account_move.py
index 1eb624f2ea5..63b3d14c5b3 100644
--- a/account_edi_simple_pdf/models/account_move.py
+++ b/account_edi_simple_pdf/models/account_move.py
@@ -60,6 +60,7 @@ def _simple_pdf_create_invoice_from_attachment(self, attachment):
                     ],
                 }
             )
+            result._onchange_partner_id()
             if result.partner_id.simple_pdf_product_id:
                 with Form(result) as invoice_form:
                     with invoice_form.invoice_line_ids.edit(0) as line_form:
diff --git a/account_edi_simple_pdf/tests/test_invoice_import.py b/account_edi_simple_pdf/tests/test_invoice_import.py
index 2f7de3e0756..2c6ba938747 100644
--- a/account_edi_simple_pdf/tests/test_invoice_import.py
+++ b/account_edi_simple_pdf/tests/test_invoice_import.py
@@ -546,3 +546,11 @@ def test_specific_python_methods(self):
         # test only pure-pdf methods
         # because we are sure they work on the Github test environment
         self._complete_import_specific_method("pypdf")
+
+    def test_test_run(self):
+        """
+        Test the test run code on the partner form
+        """
+        self.partner_ak.simple_pdf_test_file = self.ak_pdf_file_b64
+        self.partner_ak.pdf_simple_test_run()
+        self.assertIn("Current partner found", self.partner_ak.simple_pdf_test_results)