diff --git a/src/backend/langflow/interface/run.py b/src/backend/langflow/interface/run.py
index 9d3d95cda..1aff2f0e2 100644
--- a/src/backend/langflow/interface/run.py
+++ b/src/backend/langflow/interface/run.py
@@ -58,8 +58,12 @@ def get_memory_key(langchain_object):
         "chat_history": "history",
         "history": "chat_history",
     }
-    memory_key = langchain_object.memory.memory_key
-    return mem_key_dict.get(memory_key)
+    # Check if memory_key attribute exists
+    if hasattr(langchain_object.memory, 'memory_key'):
+        memory_key = langchain_object.memory.memory_key
+        return mem_key_dict.get(memory_key)
+    else:
+        return None  # or some other default value or action
 
 
 def update_memory_keys(langchain_object, possible_new_mem_key):
@@ -86,4 +90,5 @@ def update_memory_keys(langchain_object, possible_new_mem_key):
         try:
             setattr(langchain_object.memory, attr, key)
         except ValueError as exc:
-            logger.debug(f"{langchain_object.memory} has no attribute {attr} ({exc})")
+            logger.debug(
+                f"{langchain_object.memory} has no attribute {attr} ({exc})")