diff --git a/src/backend/langflow/interface/custom/custom_component.py b/src/backend/langflow/interface/custom/custom_component.py
index 5a9ddecbb..8fb5af62c 100644
--- a/src/backend/langflow/interface/custom/custom_component.py
+++ b/src/backend/langflow/interface/custom/custom_component.py
@@ -8,7 +8,7 @@ from langflow.utils import validate
 
 
 class CustomComponent(Component):
-    code: str
+    code: Optional[str]
     field_config: dict = {}
     code_class_base_inheritance = "CustomComponent"
     function_entrypoint_name = "build"