diff --git a/integration/test_rwx.html b/integration/test_rwx.html
index 79ee0f0d8a..ae6192c45f 100644
--- a/integration/test_rwx.html
+++ b/integration/test_rwx.html
@@ -40,9 +40,9 @@ <h1 class="title">Module <code>tests.test_rwx</code></h1>
 from common import wait_for_volume_creation, DATA_SIZE_IN_MB_3
 from common import create_pv_for_volume, create_pvc_for_volume
 from common import DEFAULT_STATEFULSET_TIMEOUT, DEFAULT_STATEFULSET_INTERVAL
+from common import wait_delete_pod, wait_for_pod_remount
 from common import get_core_api_client, write_pod_volume_random_data
 from common import create_pvc_spec, make_deployment_with_pvc  # NOQA
-from common import wait_for_pod_phase
 from common import core_api, statefulset, pvc, pod, client  # NOQA
 from common import RETRY_COUNTS, RETRY_INTERVAL
 from common import EXPANDED_VOLUME_SIZE
@@ -372,9 +372,10 @@ <h1 class="title">Module <code>tests.test_rwx</code></h1>
     2. Wait for StatefulSet to come up healthy.
     3. Write data and compute md5sum.
     4. Delete the share manager pod.
-    5. Check the data md5sum in statefulSet.
-    6. Write more data to it and compute md5sum.
-    7. Check the data md5sum in share manager volume.
+    5. Wait for a new pod to be created and volume getting attached.
+    6. Check the data md5sum in statefulSet.
+    7. Write more data to it and compute md5sum.
+    8. Check the data md5sum in share manager volume.
     &#34;&#34;&#34;
 
     statefulset_name = &#39;statefulset-delete-share-manager-pods-test&#39;
@@ -405,8 +406,10 @@ <h1 class="title">Module <code>tests.test_rwx</code></h1>
     delete_and_wait_pod(core_api, share_manager_name,
                         namespace=LONGHORN_NAMESPACE)
 
-    wait_for_pod_phase(core_api, share_manager_name,
-                       namespace=LONGHORN_NAMESPACE, pod_phase=&#34;Running&#34;)
+    target_pod = core_api.read_namespaced_pod(name=pod_name,
+                                              namespace=&#39;default&#39;)
+    wait_delete_pod(core_api, target_pod.metadata.uid)
+    wait_for_pod_remount(core_api, pod_name)
 
     test_data_2 = generate_random_data(VOLUME_RWTEST_SIZE)
     write_pod_volume_data(core_api, pod_name, test_data_2, filename=&#39;test2&#39;)
@@ -763,6 +766,7 @@ <h2 class="section-title" id="header-functions">Functions</h2>
 <li>Wait for StatefulSet to come up healthy.</li>
 <li>Write data and compute md5sum.</li>
 <li>Delete the share manager pod.</li>
+<li>Wait for a new pod to be created and volume getting attached.</li>
 <li>Check the data md5sum in statefulSet.</li>
 <li>Write more data to it and compute md5sum.</li>
 <li>Check the data md5sum in share manager volume.</li>
@@ -780,9 +784,10 @@ <h2 class="section-title" id="header-functions">Functions</h2>
     2. Wait for StatefulSet to come up healthy.
     3. Write data and compute md5sum.
     4. Delete the share manager pod.
-    5. Check the data md5sum in statefulSet.
-    6. Write more data to it and compute md5sum.
-    7. Check the data md5sum in share manager volume.
+    5. Wait for a new pod to be created and volume getting attached.
+    6. Check the data md5sum in statefulSet.
+    7. Write more data to it and compute md5sum.
+    8. Check the data md5sum in share manager volume.
     &#34;&#34;&#34;
 
     statefulset_name = &#39;statefulset-delete-share-manager-pods-test&#39;
@@ -813,8 +818,10 @@ <h2 class="section-title" id="header-functions">Functions</h2>
     delete_and_wait_pod(core_api, share_manager_name,
                         namespace=LONGHORN_NAMESPACE)
 
-    wait_for_pod_phase(core_api, share_manager_name,
-                       namespace=LONGHORN_NAMESPACE, pod_phase=&#34;Running&#34;)
+    target_pod = core_api.read_namespaced_pod(name=pod_name,
+                                              namespace=&#39;default&#39;)
+    wait_delete_pod(core_api, target_pod.metadata.uid)
+    wait_for_pod_remount(core_api, pod_name)
 
     test_data_2 = generate_random_data(VOLUME_RWTEST_SIZE)
     write_pod_volume_data(core_api, pod_name, test_data_2, filename=&#39;test2&#39;)