diff --git a/hadoop-ozone/dist/src/main/smoketest/balancer/testBalancer.robot b/hadoop-ozone/dist/src/main/smoketest/balancer/testBalancer.robot
index 608fd27f8e6..641bc1462bb 100644
--- a/hadoop-ozone/dist/src/main/smoketest/balancer/testBalancer.robot
+++ b/hadoop-ozone/dist/src/main/smoketest/balancer/testBalancer.robot
@@ -177,9 +177,7 @@ Verify Container Balancer for RATIS/EC containers
 
     Run Balancer Verbose Status
 
-    Sleep                   40000ms
-
-    Run Balancer Verbose History Status
+    Wait Until Keyword Succeeds    40sec    5sec   Run Balancer Verbose History Status
 
     Wait Finish Of Balancing