diff --git a/pms/tests/test_pms_checkin_partner.py b/pms/tests/test_pms_checkin_partner.py
index c91337724..c3c647f44 100644
--- a/pms/tests/test_pms_checkin_partner.py
+++ b/pms/tests/test_pms_checkin_partner.py
@@ -46,7 +46,6 @@ class TestPmsCheckinPartner(TestHotel):
         # ACTION
         self.arrange_single_checkin()
         checkins_count = len(self.reservation_1.checkin_partner_ids)
-
         # ASSERT
         self.assertEqual(
             checkins_count,
@@ -349,7 +348,7 @@ class TestPmsCheckinPartner(TestHotel):
         freezer.start()
         PmsReservation.auto_no_show()
 
-        no_show_reservations = PmsReservation.search([("state", "=", "no_show")])
+        no_show_reservations = self.folio_1.filtered(lambda r: r.state == "no_show")
 
         # ASSERT
         self.assertEqual(
@@ -371,8 +370,8 @@ class TestPmsCheckinPartner(TestHotel):
         freezer.start()
         PmsReservation.auto_no_checkout()
 
-        no_checkout_reservations = PmsReservation.search(
-            [("state", "=", "no_checkout")]
+        no_checkout_reservations = self.folio_1.filtered(
+            lambda r: r.state == "no_checkout"
         )
         freezer.stop()
         # ASSERT