1 files changed, 4 insertions, 6 deletions
diff --git a/tests/functional/glusterd/test_remove_brick_after_restart_glusterd.py b/tests/functional/glusterd/test_remove_brick_after_restart_glusterd.py
index 90e962b39..40012b009 100644
--- a/tests/functional/glusterd/test_remove_brick_after_restart_glusterd.py
+++ b/tests/functional/glusterd/test_remove_brick_after_restart_glusterd.py
@@ -29,7 +29,7 @@ from glustolibs.gluster.lib_utils import form_bricks_list
 from glustolibs.gluster.brick_ops import remove_brick
 from glustolibs.gluster.exceptions import ExecutionError
 from glustolibs.gluster.mount_ops import mount_volume, umount_volume
-from glustolibs.io.utils import validate_io_procs, wait_for_io_to_complete
+from glustolibs.io.utils import validate_io_procs
 from glustolibs.gluster.gluster_init import restart_glusterd
 
 
@@ -166,11 +166,9 @@ class TestRemoveBrickAfterRestartGlusterd(GlusterBaseClass):
         self.io_validation_complete = False
 
         # Validate IO
-        ret = validate_io_procs(self.all_mounts_procs, self.mounts)
-        self.assertTrue(ret, "IO failed on some of the clients")
-        if ret:
-            wait_for_io_to_complete(self.all_mounts_procs, self.mounts)
-            g.log.info("wait for io completed")
+        self.assertTrue(
+            validate_io_procs(self.all_mounts_procs, self.mounts),
+            "IO failed on some of the clients")
         self.io_validation_complete = True
 
         remove_brick_list = bricks_list[2:4]