diff --git a/internal/datacoord/policy_test.go b/internal/datacoord/policy_test.go
index 839228017bba5..698967ad276e9 100644
--- a/internal/datacoord/policy_test.go
+++ b/internal/datacoord/policy_test.go
@@ -467,8 +467,8 @@ func TestAssignByCountPolicySuite(t *testing.T) {
 
 func (s *AssignByCountPolicySuite) SetupSubTest() {
 	s.curCluster = []*NodeChannelInfo{
-		{1, getChannels(map[string]int64{"ch-1": 1, "ch2": 1, "ch-2": 2})},
-		{2, getChannels(map[string]int64{"ch-3": 1, "ch2": 1, "ch-4": 4})},
+		{1, getChannels(map[string]int64{"ch-1": 1, "ch-2": 1, "ch-3": 2})},
+		{2, getChannels(map[string]int64{"ch-3": 1, "ch-4": 1, "ch-5": 4})},
 		{NodeID: 3, Channels: map[string]RWChannel{}},
 	}
 }