diff --git a/tests/kvconnector/patch/0001-implement-yr-datasystem-connector-and-support-multimoda.patch b/tests/kvconnector/patch/0001-implement-yr-datasystem-connector-and-support-multimoda.patch
index 9ad5b82376846a2a7f31458c3b2e68be088588b7..6ae199e7af3477b2eca91561b32e028442ac7241 100644
--- a/tests/kvconnector/patch/0001-implement-yr-datasystem-connector-and-support-multimoda.patch
+++ b/tests/kvconnector/patch/0001-implement-yr-datasystem-connector-and-support-multimoda.patch
@@ -1392,9 +1392,9 @@ index 0000000..9b7407a
 +            if len(key_list) != len(block_id_list):
 +                logger.error(f"mget_tensors_h2d fail, request.request_id:{request.request_id}.")
 +
-+            future = self.kvc_store.mget_page_attn_blockwise_h2d(key_list, self.key_caches, block_id_list)
++            future = self.kvc_store.mget_page_attn_blockwise_h2d(key_list, self.key_caches, block_id_list, 10000)
 +            future_1 = self.kvc_store.mget_page_attn_blockwise_h2d(value_cache_key_list, self.value_caches,
-+                                                                   block_id_list)
++                                                                   block_id_list, 10000)
 +            if not self.do_async_save:
 +                get_future(future, SYNC_FUTURE_TIMEOUT)
 +                get_future(future_1, SYNC_FUTURE_TIMEOUT)