diff --git a/src/backend/base/langflow/components/prompts/Prompt.py b/src/backend/base/langflow/components/prompts/Prompt.py
index bb3b9e47c..720830fbb 100644
--- a/src/backend/base/langflow/components/prompts/Prompt.py
+++ b/src/backend/base/langflow/components/prompts/Prompt.py
@@ -1,6 +1,7 @@
 from langflow.custom import Component
 from langflow.field_typing.prompt import Prompt
 from langflow.inputs import PromptInput
+from langflow.schema.message import Message
 from langflow.template import Output
 
 
@@ -28,6 +29,6 @@ class PromptComponent(Component):
         self,
     ) -> Prompt:
         kwargs = {k: v for k, v in self._arguments.items() if k != "template"}
-        prompt = await Prompt.from_template_and_variables(self.template, kwargs)
+        prompt = await Message.from_template_and_variables(self.template, kwargs)
         self.status = prompt.format_text()
         return prompt