diff --git a/src/backend/tests/test_endpoints.py b/src/backend/tests/test_endpoints.py
index 2c4a32c8c..a57e06802 100644
--- a/src/backend/tests/test_endpoints.py
+++ b/src/backend/tests/test_endpoints.py
@@ -407,11 +407,12 @@ def test_get_vertices(client, added_flow_with_prompt_and_history, logged_in_head
     # ['ConversationBufferMemory-Lu2Nb', 'PromptTemplate-5Q0W8', 'ChatOpenAI-vy7fV', 'LLMChain-UjBh1']
     # The important part is before the - (ConversationBufferMemory, PromptTemplate, ChatOpenAI, LLMChain)
     ids = [_id.split("-")[0] for _id in response.json()["ids"]]
-    assert ids == [
+
+    assert set(ids) == {
         "ChatOpenAI",
         "PromptTemplate",
         "ConversationBufferMemory",
-    ]
+    }
 
 
 def test_build_vertex_invalid_flow_id(client, logged_in_headers):