diff --git a/internal/querynode/tsafe_replica_test.go b/internal/querynode/tsafe_replica_test.go
index c101dc60b2..cb1ada9a88 100644
--- a/internal/querynode/tsafe_replica_test.go
+++ b/internal/querynode/tsafe_replica_test.go
@@ -13,7 +13,6 @@ package querynode
 
 import (
 	"testing"
-	"time"
 
 	"github.com/stretchr/testify/assert"
 )
@@ -29,7 +28,7 @@ func TestTSafeReplica_valid(t *testing.T) {
 	timestamp := Timestamp(1000)
 	err = replica.setTSafe(defaultVChannel, defaultCollectionID, timestamp)
 	assert.NoError(t, err)
-	time.Sleep(20 * time.Millisecond)
+	<-watcher.watcherChan()
 	resT, err := replica.getTSafe(defaultVChannel)
 	assert.NoError(t, err)
 	assert.Equal(t, timestamp, resT)
@@ -49,7 +48,7 @@ func TestTSafeReplica_invalid(t *testing.T) {
 	timestamp := Timestamp(1000)
 	err = replica.setTSafe(defaultVChannel, defaultCollectionID, timestamp)
 	assert.NoError(t, err)
-	time.Sleep(20 * time.Millisecond)
+	<-watcher.watcherChan()
 	resT, err := replica.getTSafe(defaultVChannel)
 	assert.NoError(t, err)
 	assert.Equal(t, timestamp, resT)