diff --git a/libs/partners/astradb/langchain_astradb/storage.py b/libs/partners/astradb/langchain_astradb/storage.py
index 119e82c2cf4965..da4cc58593d4e2 100644
--- a/libs/partners/astradb/langchain_astradb/storage.py
+++ b/libs/partners/astradb/langchain_astradb/storage.py
@@ -60,12 +60,12 @@ async def amget(self, keys: Sequence[str]) -> List[Optional[V]]:
     def mset(self, key_value_pairs: Sequence[Tuple[str, V]]) -> None:
         self.astra_env.ensure_db_setup()
         for k, v in key_value_pairs:
-            self.collection.upsert({"_id": k, "value": self.encode_value(v)})
+            self.collection.upsert_one({"_id": k, "value": self.encode_value(v)})
 
     async def amset(self, key_value_pairs: Sequence[Tuple[str, V]]) -> None:
         await self.astra_env.aensure_db_setup()
         for k, v in key_value_pairs:
-            await self.async_collection.upsert(
+            await self.async_collection.upsert_one(
                 {"_id": k, "value": self.encode_value(v)}
             )