diff --git a/client/insert.go b/client/insert.go
index ed57c9a4..d895e5eb 100644
--- a/client/insert.go
+++ b/client/insert.go
@@ -217,6 +217,7 @@ func (c *GrpcClient) FlushV2(ctx context.Context, collName string, async bool) (
 		return nil, nil, 0, nil, err
 	}
 	channelCPs := resp.GetChannelCps()
+	flushTs := resp.GetCollFlushTs()[collName]
 	if !async {
 		segmentIDs, has := resp.GetCollSegIDs()[collName]
 		ids := segmentIDs.GetData()
@@ -224,6 +225,7 @@ func (c *GrpcClient) FlushV2(ctx context.Context, collName string, async bool) (
 			flushed := func() bool {
 				resp, err := c.Service.GetFlushState(ctx, &milvuspb.GetFlushStateRequest{
 					SegmentIDs: ids,
+					FlushTs:    flushTs,
 				})
 
 				if err != nil {