diff --git a/app/api/dao/mentorship_relation.py b/app/api/dao/mentorship_relation.py
index b18c9de1f..5b8fe99b9 100644
--- a/app/api/dao/mentorship_relation.py
+++ b/app/api/dao/mentorship_relation.py
@@ -136,12 +136,12 @@ def list_mentorship_relations(user_id=None, state=None):
             message: A message corresponding to the completed action; success if all relationships of a given user are listed, failure if otherwise.
         """
         # To check if the entered 'state' is valid.
-        valid_states = ["PENDING", "ACCEPTED", "REJECTED", "CANCELLED", "COMPLETED"]
-
         def isValidState(rel_state):
-            if rel_state in valid_states:
-                return True
-            return False
+            try:
+                MentorshipRelationState[rel_state]
+            except KeyError:
+                return False
+            return True
 
         user = UserModel.find_by_id(user_id)
         all_relations = user.mentor_relations + user.mentee_relations