diff --git a/src/backend/langflow/interface/importing/utils.py b/src/backend/langflow/interface/importing/utils.py
index cf804e67d..1176de07c 100644
--- a/src/backend/langflow/interface/importing/utils.py
+++ b/src/backend/langflow/interface/importing/utils.py
@@ -181,7 +181,7 @@ class MyPythonClass:
             llm=llm, verbose=True, memory=ConversationBufferMemory()
         )
 
-    def build(self, name: str) -> ConversationChain:
+    def build(self, my_custom_input: str) -> ConversationChain:
         conversation = self.my_conversation()
         
         return conversation
diff --git a/src/backend/langflow/utils/constants.py b/src/backend/langflow/utils/constants.py
index 8cec38c6e..a7de648b3 100644
--- a/src/backend/langflow/utils/constants.py
+++ b/src/backend/langflow/utils/constants.py
@@ -62,7 +62,7 @@ class MyPythonClass:
             llm=llm, verbose=True, memory=ConversationBufferMemory()
         )
 
-    def build(self, name: str) -> ConversationChain:
+    def build(self, my_custom_input: str) -> ConversationChain:
         conversation = self.my_conversation()
         
         return conversation