diff --git a/src/backend/langflow/__init__.py b/src/backend/langflow/__init__.py
index d6c645486..5920369e2 100644
--- a/src/backend/langflow/__init__.py
+++ b/src/backend/langflow/__init__.py
@@ -1,7 +1,7 @@
 from importlib import metadata
 from langflow.cache import cache_manager
 from langflow.processing.process import load_flow_from_json
-from langflow.utils.types import Prompt
+from langflow.interface.custom.custom_component import CustomComponent
 
 try:
     __version__ = metadata.version(__package__)
@@ -10,4 +10,4 @@ except metadata.PackageNotFoundError:
     __version__ = ""
 del metadata  # optional, avoids polluting the results of dir(__package__)
 
-__all__ = ["load_flow_from_json", "cache_manager", "Prompt"]
+__all__ = ["load_flow_from_json", "cache_manager", "CustomComponent"]
diff --git a/src/backend/langflow/interface/custom/constants.py b/src/backend/langflow/interface/custom/constants.py
index 00aa5314d..8e5db39b8 100644
--- a/src/backend/langflow/interface/custom/constants.py
+++ b/src/backend/langflow/interface/custom/constants.py
@@ -35,8 +35,7 @@ CUSTOM_COMPONENT_SUPPORTED_TYPES = {
 
 
 DEFAULT_CUSTOM_COMPONENT_CODE = """
-from langflow import Prompt
-from langflow.interface.custom.custom_component import CustomComponent
+from langflow import CustomComponent
 
 from langchain.llms.base import BaseLLM
 from langchain.chains import LLMChain
@@ -48,11 +47,12 @@ import requests
 class YourComponent(CustomComponent):
     display_name: str = "Your Component"
     description: str = "Your description"
-    field_config = { "url": { "multiline": True, "required": True } }
 
-    def build(self, url: str, llm: BaseLLM, template: Prompt) -> Document:
+    def build_config(self):
+        return { "url": { "multiline": True, "required": True } }
+
+    def build(self, url: str, llm: BaseLLM, prompt: PromptTemplate) -> Document:
         response = requests.get(url)
-        prompt = PromptTemplate.from_template(template)
         chain = LLMChain(llm=llm, prompt=prompt)
         result = chain.run(response.text[:300])
         return Document(page_content=str(result))