diff --git a/tests/integration/test_storage_postgresql_replica/test.py b/tests/integration/test_storage_postgresql_replica/test.py
index d2ac5d1a109..aca33816d75 100644
--- a/tests/integration/test_storage_postgresql_replica/test.py
+++ b/tests/integration/test_storage_postgresql_replica/test.py
@@ -50,20 +50,17 @@ queries = [
 def check_tables_are_synchronized(
     table_name, order_by="key", postgres_database="postgres_database"
 ):
-    expected = instance.query(
-        "select * from {}.{} order by {};".format(
-            postgres_database, table_name, order_by
+    while True:
+        expected = instance.query(
+            "select * from {}.{} order by {};".format(
+                postgres_database, table_name, order_by
+            )
         )
-    )
-    result = instance.query(
-        "select * from test.{} order by {};".format(table_name, order_by)
-    )
-
-    while result != expected:
-        time.sleep(0.5)
         result = instance.query(
             "select * from test.{} order by {};".format(table_name, order_by)
         )
+        if result == expected:
+            break
 
     assert result == expected
 
@@ -103,7 +100,7 @@ def create_clickhouse_postgres_db(ip, port, name="postgres_database"):
     )
 
 
-def create_materialized_table(ip, port, table_name='postgresql_replica'):
+def create_materialized_table(ip, port, table_name="postgresql_replica"):
     instance.query(
         f"""
         CREATE TABLE test.{table_name} (key Int64, value Int64)
@@ -709,12 +706,12 @@ def test_abrupt_server_restart_while_heavy_replication(started_cluster):
     table_name = "postgresql_replica_697"
     create_postgres_table(cursor, table_name)
 
-    instance.query(
-        f"INSERT INTO postgres_database.{table_name} SELECT -1, 1"
-    )
+    instance.query(f"INSERT INTO postgres_database.{table_name} SELECT -1, 1")
     instance.query(f"DROP TABLE IF EXISTS test.{table_name} NO DELAY")
     create_materialized_table(
-        ip=started_cluster.postgres_ip, port=started_cluster.postgres_port, table_name=table_name
+        ip=started_cluster.postgres_ip,
+        port=started_cluster.postgres_port,
+        table_name=table_name,
     )
 
     n = 1
@@ -722,7 +719,7 @@ def test_abrupt_server_restart_while_heavy_replication(started_cluster):
         sleep(1)
         n += 1
         if n > 10:
-            break;
+            break
 
     for query in queries:
         cursor.execute(query.format(table_name))