diff --git a/src/backend/base/langflow/custom/custom_component/component.py b/src/backend/base/langflow/custom/custom_component/component.py
index aaa16771d..6e1b2e723 100644
--- a/src/backend/base/langflow/custom/custom_component/component.py
+++ b/src/backend/base/langflow/custom/custom_component/component.py
@@ -17,7 +17,7 @@ import yaml
 from loguru import logger
 from pydantic import BaseModel
 
-from langflow.schema.schema import Record
+from langflow.schema.record import Record
 from langflow.template.field.base import Input, Output
 
 from .custom_component import CustomComponent