diff --git a/libs/standard-tests/langchain_tests/integration_tests/base_store.py b/libs/standard-tests/langchain_tests/integration_tests/base_store.py
index cba4c245e1d..e7f6812901d 100644
--- a/libs/standard-tests/langchain_tests/integration_tests/base_store.py
+++ b/libs/standard-tests/langchain_tests/integration_tests/base_store.py
@@ -239,7 +239,7 @@ class BaseStoreAsyncTests(BaseStandardTests, Generic[V]):
         await kv_store.amset(key_value_pairs)
         await kv_store.amset(key_value_pairs)
         assert await kv_store.amget(["foo", "bar"]) == [foo, bar]
-        assert sorted(kv_store.yield_keys()) == ["bar", "foo"]
+        assert sorted([key async for key in kv_store.ayield_keys()]) == ["bar", "foo"]
 
     async def test_get_can_get_same_value(
         self, kv_store: BaseStore[str, V], three_values: tuple[V, V, V]