diff --git a/src/backend/langflow/interface/custom/constants.py b/src/backend/langflow/interface/custom/constants.py
index 298b7a969..36ebf420e 100644
--- a/src/backend/langflow/interface/custom/constants.py
+++ b/src/backend/langflow/interface/custom/constants.py
@@ -1,5 +1,5 @@
 DEFAULT_CUSTOM_COMPONENT_CODE = """from langflow import CustomComponent
-
+from typing import Optional, List, Dict, Union
 from langflow.field_typing import (
     AgentExecutor,
     BaseChatMemory,