diff --git a/internal/querynode/collection.go b/internal/querynode/collection.go
index 639b961545..2f2fa6e32a 100644
--- a/internal/querynode/collection.go
+++ b/internal/querynode/collection.go
@@ -52,7 +52,7 @@ func (c *Collection) addPartitionID(partitionID UniqueID) {
 func (c *Collection) removePartitionID(partitionID UniqueID) {
 	tmpIDs := make([]UniqueID, 0)
 	for _, id := range c.partitionIDs {
-		if id == partitionID {
+		if id != partitionID {
 			tmpIDs = append(tmpIDs, id)
 		}
 	}
diff --git a/internal/querynode/partition.go b/internal/querynode/partition.go
index cfb6003a1a..95f6cc6c38 100644
--- a/internal/querynode/partition.go
+++ b/internal/querynode/partition.go
@@ -46,7 +46,7 @@ func (p *Partition) addSegmentID(segmentID UniqueID) {
 func (p *Partition) removeSegmentID(segmentID UniqueID) {
 	tmpIDs := make([]UniqueID, 0)
 	for _, id := range p.segmentIDs {
-		if id == segmentID {
+		if id != segmentID {
 			tmpIDs = append(tmpIDs, id)
 		}
 	}