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):