diff --git a/internal/datanode/compaction/merge_sort.go b/internal/datanode/compaction/merge_sort.go
index bcf8dd8bb7..a43cc0c179 100644
--- a/internal/datanode/compaction/merge_sort.go
+++ b/internal/datanode/compaction/merge_sort.go
@@ -105,6 +105,12 @@ func mergeSortMultipleSegments(ctx context.Context,
 				deletedRowCount++
 				continue
 			}
+
+			// Filtering expired entity
+			if isExpiredEntity(collectionTtl, currentTs, typeutil.Timestamp(v.Timestamp)) {
+				expiredRowCount++
+				continue
+			}
 			return v, nil
 		}
 	}
@@ -125,12 +131,6 @@ func mergeSortMultipleSegments(ctx context.Context,
 		smallest := heap.Pop(&pq).(*PQItem)
 		v := smallest.Value
 
-		// Filtering expired entity
-		if isExpiredEntity(collectionTtl, currentTs, typeutil.Timestamp(v.Timestamp)) {
-			expiredRowCount++
-			continue
-		}
-
 		err := mWriter.Write(v)
 		if err != nil {
 			log.Warn("compact wrong, failed to writer row", zap.Error(err))