diff --git a/integration/test_ha.html b/integration/test_ha.html
index 033009a516..e0c2ccfdff 100644
--- a/integration/test_ha.html
+++ b/integration/test_ha.html
@@ -2012,7 +2012,7 @@ <h1 class="title">Module <code>tests.test_ha</code></h1>
     wait_for_volume_replica_count(client, volume_name, 3)
 
     volume = client.by_id_volume(volume_name)
-    assert volume.robustness == &#34;healthy&#34;
+    wait_for_volume_healthy(client, volume_name)
 
     check_volume_data(volume, data)
 
@@ -5777,7 +5777,7 @@ <h2 class="section-title" id="header-functions">Functions</h2>
     wait_for_volume_replica_count(client, volume_name, 3)
 
     volume = client.by_id_volume(volume_name)
-    assert volume.robustness == &#34;healthy&#34;
+    wait_for_volume_healthy(client, volume_name)
 
     check_volume_data(volume, data)</code></pre>
 </details>