diff --git a/src/backend/base/langflow/utils/validate.py b/src/backend/base/langflow/utils/validate.py
index 0871dbd82..c3bbc9df8 100644
--- a/src/backend/base/langflow/utils/validate.py
+++ b/src/backend/base/langflow/utils/validate.py
@@ -159,7 +159,10 @@ def create_class(code, class_name):
 
     # Replace from langflow import CustomComponent with from langflow.custom import CustomComponent
     code = code.replace("from langflow import CustomComponent", "from langflow.custom import CustomComponent")
-
+    code = code.replace(
+        "from langflow.interface.custom.custom_component import CustomComponent",
+        "from langflow.custom import CustomComponent",
+    )
     module = ast.parse(code)
     exec_globals = prepare_global_scope(code, module)