diff --git a/src/apify/memory_storage/resource_clients/key_value_store.py b/src/apify/memory_storage/resource_clients/key_value_store.py
index 6c766483..95e1ab12 100644
--- a/src/apify/memory_storage/resource_clients/key_value_store.py
+++ b/src/apify/memory_storage/resource_clients/key_value_store.py
@@ -7,7 +7,7 @@
 import warnings
 from datetime import datetime
 from operator import itemgetter
-from typing import TYPE_CHECKING, Any, Dict, Optional, Union
+from typing import TYPE_CHECKING, Any, AsyncIterator, Dict, Optional, Union
 
 import aioshutil