diff --git a/src/backend/langflow/graph/graph/constants.py b/src/backend/langflow/graph/graph/constants.py
index 40d77e31c..a2fd287eb 100644
--- a/src/backend/langflow/graph/graph/constants.py
+++ b/src/backend/langflow/graph/graph/constants.py
@@ -19,18 +19,18 @@ from typing import Dict, Type
 
 
 VERTEX_TYPE_MAP: Dict[str, Type[Vertex]] = {
-    **{t: types.types.PromptVertex for t in prompt_creator.to_list()},
-    **{t: types.types.AgentVertex for t in agent_creator.to_list()},
-    **{t: types.types.ChainVertex for t in chain_creator.to_list()},
-    **{t: types.types.ToolVertex for t in tool_creator.to_list()},
-    **{t: types.types.ToolkitVertex for t in toolkits_creator.to_list()},
-    **{t: types.types.WrapperVertex for t in wrapper_creator.to_list()},
-    **{t: types.types.LLMVertex for t in llm_creator.to_list()},
-    **{t: types.types.MemoryVertex for t in memory_creator.to_list()},
-    **{t: types.types.EmbeddingVertex for t in embedding_creator.to_list()},
-    **{t: types.types.VectorStoreVertex for t in vectorstore_creator.to_list()},
-    **{t: types.types.DocumentLoaderVertex for t in documentloader_creator.to_list()},
-    **{t: types.types.TextSplitterVertex for t in textsplitter_creator.to_list()},
+    **{t: types.PromptVertex for t in prompt_creator.to_list()},
+    **{t: types.AgentVertex for t in agent_creator.to_list()},
+    **{t: types.ChainVertex for t in chain_creator.to_list()},
+    **{t: types.ToolVertex for t in tool_creator.to_list()},
+    **{t: types.ToolkitVertex for t in toolkits_creator.to_list()},
+    **{t: types.WrapperVertex for t in wrapper_creator.to_list()},
+    **{t: types.LLMVertex for t in llm_creator.to_list()},
+    **{t: types.MemoryVertex for t in memory_creator.to_list()},
+    **{t: types.EmbeddingVertex for t in embedding_creator.to_list()},
+    **{t: types.VectorStoreVertex for t in vectorstore_creator.to_list()},
+    **{t: types.DocumentLoaderVertex for t in documentloader_creator.to_list()},
+    **{t: types.TextSplitterVertex for t in textsplitter_creator.to_list()},
     **{t: types.OutputParserVertex for t in output_parser_creator.to_list()},
     **{t: types.RetrieverVertex for t in retriever_creator.to_list()},
 }