fix existing test

unicef · Dec 3, 2024 · bcd6b14 · bcd6b14
1 parent c569868
commit bcd6b14
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/unit/apps/grievance/test_reassign_roles_services.py b/tests/unit/apps/grievance/test_reassign_roles_services.py
@@ -10,9 +10,9 @@
 )
 from hct_mis_api.apps.household.fixtures import HouseholdFactory, IndividualFactory
 from hct_mis_api.apps.household.models import (
+    RELATIONSHIP_UNKNOWN,
     ROLE_ALTERNATE,
     ROLE_PRIMARY,
-    UNKNOWN,
     Individual,
     IndividualRoleInHousehold,
 )
@@ -97,7 +97,7 @@ def test_reassign_roles_on_marking_as_duplicate_individual_service(self) -> None
         self.household.refresh_from_db()
         self.assertEqual(self.household.head_of_household, self.no_role_individual)
         for individual in self.household.individuals.exclude(id=self.no_role_individual.id):
-            self.assertEqual(individual.relationship, UNKNOWN)
+            self.assertEqual(individual.relationship, RELATIONSHIP_UNKNOWN)
 
     def test_reassign_roles_on_marking_as_duplicate_individual_service_wrong_program(self) -> None:
         program_two = ProgramFactory(name="Test program TWO", business_area=self.business_area)