diff --git a/contract_digitized_signature/tests/test_contract_tracking.py b/contract_digitized_signature/tests/test_contract_tracking.py
index 5ffe6125d..729d8b1aa 100644
--- a/contract_digitized_signature/tests/test_contract_tracking.py
+++ b/contract_digitized_signature/tests/test_contract_tracking.py
@@ -9,14 +9,18 @@ class TestContractSignatureTracking(common.SavepointCase):
     def setUpClass(cls):
         super(TestContractSignatureTracking, cls).setUpClass()
         # Simple 1x1 transparent base64 encoded GIF
-        cls.image = 'R0lGODlhAQABAAAAACH5BAEKAAEALAAAAAABAAEAAAICTAEAOw=='
+        cls.image = "R0lGODlhAQABAAAAACH5BAEKAAEALAAAAAABAAEAAAICTAEAOw=="
 
     def test_contract_signature_tracking(self):
-        self.contract = self.env['contract.contract'].create({
-            'name': 'Test Contract',
-            'customer_signature': self.image,
-        })
-        message = self.env['mail.message'].search([
-            ('res_id', '=', self.contract.id)
-        ], order='id desc', limit=1)
-        self.assertIn('Signature has been created.', message.body)
+        partner = self.env["res.partner"].create({"name": "test partner"})
+        self.contract = self.env["contract.contract"].create(
+            {
+                "name": "Test Contract",
+                "customer_signature": self.image,
+                "partner_id": partner.id,
+            }
+        )
+        message = self.env["mail.message"].search(
+            [("res_id", "=", self.contract.id)], order="id desc", limit=1
+        )
+        self.assertIn("Signature has been created.", message.body)