diff --git a/apps/emqx_durable_storage/test/emqx_ds_replication_SUITE.erl b/apps/emqx_durable_storage/test/emqx_ds_replication_SUITE.erl
index d1f98b3c3..5ff1d5fb2 100644
--- a/apps/emqx_durable_storage/test/emqx_ds_replication_SUITE.erl
+++ b/apps/emqx_durable_storage/test/emqx_ds_replication_SUITE.erl
@@ -52,7 +52,7 @@ t_replication_transfers_snapshots(Config) ->
     ?retry(
         500,
         10,
-        ?assertMatch([_], shards_online(Node, ?DB))
+        ?assertMatch([[_], [_], [_]], [shards_online(N, ?DB) || N <- Nodes])
     ),
 
     %% Stop the DB on the "offline" node.