From f5cebcaf6afd163511dcdd77a2f7a87a9439fae2 Mon Sep 17 00:00:00 2001
From: Alexey Masterov <alexey.masterov@databricks.com>
Date: Thu, 17 Jul 2025 12:34:43 +0200
Subject: [PATCH] Wait for the snapshot to complete

---
 test_runner/random_ops/test_random_ops.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/test_runner/random_ops/test_random_ops.py b/test_runner/random_ops/test_random_ops.py
index 35a657ec56..e2a9ace881 100644
--- a/test_runner/random_ops/test_random_ops.py
+++ b/test_runner/random_ops/test_random_ops.py
@@ -481,8 +481,8 @@ class NeonProject:
                         snapshot_name,
                     ),
                 )
+                self.wait()
                 cur.execute("UPDATE sanity_check SET value = 'tainted' || value")
-        self.wait()
         return snapshot
 
     def delete_snapshot(self, snapshot_id: str) -> None: