diff --git a/internal/datanode/flow_graph_delete_node_test.go b/internal/datanode/flow_graph_delete_node_test.go
index 717ac0bf20..40264f0320 100644
--- a/internal/datanode/flow_graph_delete_node_test.go
+++ b/internal/datanode/flow_graph_delete_node_test.go
@@ -156,7 +156,7 @@ func Test_GetSegmentsByPKs(t *testing.T) {
 		4: {4, 5},
 	}
 	for key, value := range expected {
-		assert.EqualValues(t, value, results[key])
+		assert.ElementsMatch(t, value, results[key])
 	}
 
 }