diff --git a/src/backend/base/langflow/base/models/model.py b/src/backend/base/langflow/base/models/model.py
index 74d350446..a8069dcf9 100644
--- a/src/backend/base/langflow/base/models/model.py
+++ b/src/backend/base/langflow/base/models/model.py
@@ -5,11 +5,11 @@ from langchain_core.language_models.chat_models import BaseChatModel
 from langchain_core.language_models.llms import LLM
 from langchain_core.messages import AIMessage, HumanMessage, SystemMessage
 
-from langflow.custom import CustomComponent
+from langflow.custom import Component
 from langflow.field_typing.prompt import Prompt
 
 
-class LCModelComponent(CustomComponent):
+class LCModelComponent(Component):
     display_name: str = "Model Name"
     description: str = "Model Description"