diff --git a/server/src/internalClusterTest/java/org/opensearch/indices/replication/SegmentReplicationIT.java b/server/src/internalClusterTest/java/org/opensearch/indices/replication/SegmentReplicationIT.java
index 51c0c8710d39d..59c1c45ccd3ea 100644
--- a/server/src/internalClusterTest/java/org/opensearch/indices/replication/SegmentReplicationIT.java
+++ b/server/src/internalClusterTest/java/org/opensearch/indices/replication/SegmentReplicationIT.java
@@ -738,6 +738,7 @@ public void testReplicaHasDiffFilesThanPrimary() throws Exception {
 
         final SegmentInfos segmentInfos = SegmentInfos.readLatestCommit(replicaShard.store().directory());
         replicaShard.finalizeReplication(segmentInfos);
+        ensureYellow(INDEX_NAME);
 
         final int docCount = scaledRandomIntBetween(10, 200);
         for (int i = 0; i < docCount; i++) {
@@ -745,7 +746,8 @@ public void testReplicaHasDiffFilesThanPrimary() throws Exception {
             refresh(INDEX_NAME);
         }
         // Refresh, this should trigger round of segment replication
-        assertBusy(() -> { assertDocCounts(docCount, replicaNode); });
+        waitForSearchableDocs(docCount, primaryNode, replicaNode);
+        verifyStoreContent();
         final IndexShard replicaAfterFailure = getIndexShard(replicaNode, INDEX_NAME);
         assertNotEquals(replicaAfterFailure.routingEntry().allocationId().getId(), replicaShard.routingEntry().allocationId().getId());
     }