diff --git a/src/backend/base/langflow/components/memories/AstraDBMessageReader.py b/src/backend/base/langflow/components/memories/AstraDBMessageReader.py
index c61909887..8d845be73 100644
--- a/src/backend/base/langflow/components/memories/AstraDBMessageReader.py
+++ b/src/backend/base/langflow/components/memories/AstraDBMessageReader.py
@@ -1,8 +1,7 @@
 from typing import Optional, cast
 
-
 from langflow.base.memory.memory import BaseMemoryComponent
-from langflow.schema.schema import Record
+from langflow.schema.record import Record
 
 
 class AstraDBMessageReaderComponent(BaseMemoryComponent):
diff --git a/src/backend/base/langflow/components/memories/AstraDBMessageWriter.py b/src/backend/base/langflow/components/memories/AstraDBMessageWriter.py
index c63f27db8..7aae7009b 100644
--- a/src/backend/base/langflow/components/memories/AstraDBMessageWriter.py
+++ b/src/backend/base/langflow/components/memories/AstraDBMessageWriter.py
@@ -1,12 +1,9 @@
 from typing import Optional
 
-from langchain_astradb import AstraDBChatMessageHistory
 from langchain_core.messages import BaseMessage
 
 from langflow.base.memory.memory import BaseMemoryComponent
-from langflow.schema.schema import Record
-
-from langchain_core.messages import BaseMessage
+from langflow.schema.record import Record
 
 
 class AstraDBMessageWriterComponent(BaseMemoryComponent):
diff --git a/src/backend/base/langflow/components/memories/CassandraMessageReader.py b/src/backend/base/langflow/components/memories/CassandraMessageReader.py
index 3545d444a..3fd11d772 100644
--- a/src/backend/base/langflow/components/memories/CassandraMessageReader.py
+++ b/src/backend/base/langflow/components/memories/CassandraMessageReader.py
@@ -3,7 +3,7 @@ from typing import Optional, cast
 from langchain_community.chat_message_histories import CassandraChatMessageHistory
 
 from langflow.base.memory.memory import BaseMemoryComponent
-from langflow.schema.schema import Record
+from langflow.schema.record import Record
 
 
 class CassandraMessageReaderComponent(BaseMemoryComponent):
diff --git a/src/backend/base/langflow/components/memories/CassandraMessageWriter.py b/src/backend/base/langflow/components/memories/CassandraMessageWriter.py
index 716364b64..c8e3831a5 100644
--- a/src/backend/base/langflow/components/memories/CassandraMessageWriter.py
+++ b/src/backend/base/langflow/components/memories/CassandraMessageWriter.py
@@ -1,10 +1,10 @@
 from typing import Optional
 
-from langflow.base.memory.memory import BaseMemoryComponent
-from langflow.schema.schema import Record
-
-from langchain_core.messages import BaseMessage
 from langchain_community.chat_message_histories import CassandraChatMessageHistory
+from langchain_core.messages import BaseMessage
+
+from langflow.base.memory.memory import BaseMemoryComponent
+from langflow.schema.record import Record
 
 
 class CassandraMessageWriterComponent(BaseMemoryComponent):
diff --git a/src/backend/base/langflow/components/prompts/Prompt.py b/src/backend/base/langflow/components/prompts/Prompt.py
index ca421b395..d2d9f78a0 100644
--- a/src/backend/base/langflow/components/prompts/Prompt.py
+++ b/src/backend/base/langflow/components/prompts/Prompt.py
@@ -19,6 +19,6 @@ class PromptComponent(CustomComponent):
         template: Prompt,
         **kwargs,
     ) -> Prompt:
-        prompt = await Prompt.from_template_and_variables(template, kwargs)
+        prompt = await Prompt.from_template_and_variables(template, kwargs)  # type: ignore
         self.status = prompt.format_text()
         return prompt
diff --git a/src/backend/base/langflow/components/tools/PythonCodeStructuredTool.py b/src/backend/base/langflow/components/tools/PythonCodeStructuredTool.py
index cf9d85974..84392cbdb 100644
--- a/src/backend/base/langflow/components/tools/PythonCodeStructuredTool.py
+++ b/src/backend/base/langflow/components/tools/PythonCodeStructuredTool.py
@@ -91,4 +91,4 @@ class PythonCodeStructuredTool(CustomComponent):
         tool = StructuredTool.from_function(
             func=func, args_schema=_class, name=name, description=description, return_direct=return_direct
         )
-        return tool
+        return tool  # type: ignore