diff --git a/pms/tests/test_pms_reservation.py b/pms/tests/test_pms_reservation.py
index ed1a50f11a..faab69ea6b 100644
--- a/pms/tests/test_pms_reservation.py
+++ b/pms/tests/test_pms_reservation.py
@@ -1679,7 +1679,7 @@ def test_reservation_action_checkout(self):
                 "name": "30065089H",
                 "valid_from": datetime.date.today(),
                 "partner_id": host.id,
-                "document_country_id": self.env.ref("base.es").id,
+                "country_id": self.env.ref("base.es").id,
             }
         )
         r1 = self.env["pms.reservation"].create(
@@ -1938,7 +1938,7 @@ def test_compute_ready_for_checkin(self):
                 "name": "30065000H",
                 "valid_from": datetime.date.today(),
                 "partner_id": self.host1.id,
-                "document_country_id": self.env.ref("base.es").id,
+                "country_id": self.env.ref("base.es").id,
             }
         )
         self.host2 = self.env["res.partner"].create(
@@ -1956,7 +1956,7 @@ def test_compute_ready_for_checkin(self):
                 "name": "30065089H",
                 "valid_from": datetime.date.today(),
                 "partner_id": self.host2.id,
-                "document_country_id": self.env.ref("base.es").id,
+                "country_id": self.env.ref("base.es").id,
             }
         )
         self.reservation = self.env["pms.reservation"].create(
diff --git a/pms_l10n_es/tests/test_wizard_ine.py b/pms_l10n_es/tests/test_wizard_ine.py
index df15c71c9f..3b0a6e1efb 100644
--- a/pms_l10n_es/tests/test_wizard_ine.py
+++ b/pms_l10n_es/tests/test_wizard_ine.py
@@ -116,6 +116,7 @@ def ideal_scenario(self):
                 "name": "55103354T",
                 "valid_from": datetime.date.today(),
                 "partner_id": self.partner_1.id,
+                "country_id": self.country_italy.id,
             }
         )
 
@@ -136,6 +137,7 @@ def ideal_scenario(self):
                 "name": "45437298Q",
                 "valid_from": datetime.date.today(),
                 "partner_id": self.partner_2.id,
+                "country_id": self.country_russia.id,
             }
         )
         # Create partner 3 (italy)
@@ -155,6 +157,7 @@ def ideal_scenario(self):
                 "name": "81534086Y",
                 "valid_from": datetime.date.today(),
                 "partner_id": self.partner_3.id,
+                "country_id": self.country_italy.id,
             }
         )
         # Create partner 4 (italy)
@@ -174,6 +177,7 @@ def ideal_scenario(self):
                 "name": "00807643K",
                 "valid_from": datetime.date.today(),
                 "partner_id": self.partner_4.id,
+                "country_id": self.country_italy.id,
             }
         )
         # Create partner 5 (afghanistan)
@@ -193,6 +197,7 @@ def ideal_scenario(self):
                 "name": "54564399G",
                 "valid_from": datetime.date.today(),
                 "partner_id": self.partner_5.id,
+                "country_id": self.country_afghanistan.id,
             }
         )
         # Create partner 6 (afghanistan)
@@ -212,6 +217,7 @@ def ideal_scenario(self):
                 "name": "39854152M",
                 "valid_from": datetime.date.today(),
                 "partner_id": self.partner_6.id,
+                "country_id": self.country_afghanistan.id,
             }
         )
         # Create partner 7 (afghanistan)
@@ -231,6 +237,7 @@ def ideal_scenario(self):
                 "name": "39854152O",
                 "valid_from": datetime.date.today(),
                 "partner_id": self.partner_7.id,
+                "country_id": self.country_afghanistan.id,
             }
         )
 
@@ -449,6 +456,7 @@ def pending_checkins_scenario(self):
                 "name": "15103354T",
                 "valid_from": datetime.date.today(),
                 "partner_id": self.partner_russia_1.id,
+                "country_id": self.country_russia.id,
             }
         )
         self.env["res.partner.id_number"].create(
@@ -457,6 +465,7 @@ def pending_checkins_scenario(self):
                 "name": "25103354T",
                 "valid_from": datetime.date.today(),
                 "partner_id": self.partner_russia_2.id,
+                "country_id": self.country_russia.id,
             }
         )
         self.env["res.partner.id_number"].create(
@@ -465,6 +474,7 @@ def pending_checkins_scenario(self):
                 "name": "35103354T",
                 "valid_from": datetime.date.today(),
                 "partner_id": self.partner_russia_3.id,
+                "country_id": self.country_russia.id,
             }
         )