diff --git a/tests/integration/test_merge_tree_azure_blob_storage/test.py b/tests/integration/test_merge_tree_azure_blob_storage/test.py
index 86b70f8db70..dcca85fca7a 100644
--- a/tests/integration/test_merge_tree_azure_blob_storage/test.py
+++ b/tests/integration/test_merge_tree_azure_blob_storage/test.py
@@ -7,6 +7,7 @@ import pytest
 from helpers.cluster import ClickHouseCluster
 from helpers.utility import generate_values, replace_config, SafeThread
 from azure.storage.blob import BlobServiceClient
+from test_storage_azure_blob_storage.test import azure_query
 
 
 SCRIPT_DIR = os.path.dirname(os.path.realpath(__file__))
@@ -38,28 +39,10 @@ def cluster():
         cluster.shutdown()
 
 
-# Note: use this for selects and inserts and create table queries.
+# Note: use azure_query for selects and inserts and create table queries.
 # For inserts there is no guarantee that retries will not result in duplicates.
-# But it is better to retry anyway because 'Connection was closed by the server' error
+# But it is better to retry anyway because connection related errors
 # happens in fact only for inserts because reads already have build-in retries in code.
-def azure_query(node, query, try_num=3, settings={}):
-    for i in range(try_num):
-        try:
-            return node.query(query, settings=settings)
-        except Exception as ex:
-            retriable_errors = [
-                "DB::Exception: Azure::Core::Http::TransportException: Connection was closed by the server while trying to read a response"
-            ]
-            retry = False
-            for error in retriable_errors:
-                if error in str(ex):
-                    retry = True
-                    logging.info(f"Try num: {i}. Having retriable error: {ex}")
-                    break
-            if not retry or i == try_num - 1:
-                raise Exception(ex)
-            continue
-
 
 def create_table(node, table_name, **additional_settings):
     settings = {
diff --git a/tests/integration/test_storage_azure_blob_storage/test.py b/tests/integration/test_storage_azure_blob_storage/test.py
index 21f57a67495..6e103bdf96d 100644
--- a/tests/integration/test_storage_azure_blob_storage/test.py
+++ b/tests/integration/test_storage_azure_blob_storage/test.py
@@ -43,6 +43,8 @@ def azure_query(node, query, try_num=10, settings={}):
             retriable_errors = [
                 "DB::Exception: Azure::Core::Http::TransportException: Connection was closed by the server while trying to read a response",
                 "DB::Exception: Azure::Core::Http::TransportException: Connection closed before getting full response or response is less than expected",
+                "DB::Exception: Azure::Core::Http::TransportException: Connection was closed by the server while trying to read a response",
+                "DB::Exception: Azure::Core::Http::TransportException: Error while polling for socket ready read",
             ]
             retry = False
             for error in retriable_errors: