diff --git a/src/backend/langflow/interface/custom/constants.py b/src/backend/langflow/interface/custom/constants.py
index d3e8dd3a9..298b7a969 100644
--- a/src/backend/langflow/interface/custom/constants.py
+++ b/src/backend/langflow/interface/custom/constants.py
@@ -1,23 +1,27 @@
 DEFAULT_CUSTOM_COMPONENT_CODE = """from langflow import CustomComponent
 
 from langflow.field_typing import (
-    Tool,
-    PromptTemplate,
-    Chain,
+    AgentExecutor,
     BaseChatMemory,
+    BaseLanguageModel,
     BaseLLM,
     BaseLoader,
     BaseMemory,
     BaseOutputParser,
+    BasePromptTemplate,
     BaseRetriever,
-    VectorStore,
-    Embeddings,
-    TextSplitter,
-    Document,
-    AgentExecutor,
-    NestedDict,
+    Callable,
+    Chain,
+    ChatPromptTemplate,
     Data,
+    Document,
+    Embeddings,
+    NestedDict,
     Object,
+    PromptTemplate,
+    TextSplitter,
+    Tool,
+    VectorStore,
 )