diff --git a/src/backend/base/langflow/utils/validate.py b/src/backend/base/langflow/utils/validate.py
index e7bd4ae05..b5363dc8e 100644
--- a/src/backend/base/langflow/utils/validate.py
+++ b/src/backend/base/langflow/utils/validate.py
@@ -157,6 +157,9 @@ def create_class(code, class_name):
     if not hasattr(ast, "TypeIgnore"):
         ast.TypeIgnore = create_type_ignore_class()
 
+    # Replace from langflow import CustomComponent with from langflow.custom import CustomComponent
+    code = code.replace("from langflow import CustomComponent", "from langflow.custom import CustomComponent")
+
     module = ast.parse(code)
     exec_globals = prepare_global_scope(code, module)