diff --git a/weed/mq/pub_balancer/allocate_test.go b/weed/mq/pub_balancer/allocate_test.go
index 7e7a09dab..4b36af388 100644
--- a/weed/mq/pub_balancer/allocate_test.go
+++ b/weed/mq/pub_balancer/allocate_test.go
@@ -97,9 +97,7 @@ func TestEnsureAssignmentsToActiveBrokersX(t *testing.T) {
 					{
 						LeaderBroker: "",
 						Partition:    &mq_pb.Partition{},
-						FollowerBrokers: []string{
-							"localhost:2",
-						},
+						FollowerBroker: "localhost:2",
 					},
 				},
 			},
@@ -114,9 +112,7 @@ func TestEnsureAssignmentsToActiveBrokersX(t *testing.T) {
 					{
 						LeaderBroker: "localhost:1",
 						Partition:    &mq_pb.Partition{},
-						FollowerBrokers: []string{
-							"",
-						},
+						FollowerBroker: "",
 					},
 				},
 			},
@@ -131,9 +127,7 @@ func TestEnsureAssignmentsToActiveBrokersX(t *testing.T) {
 					{
 						LeaderBroker: "localhost:1",
 						Partition:    &mq_pb.Partition{},
-						FollowerBrokers: []string{
-							"localhost:200",
-						},
+						FollowerBroker: "localhost:200",
 					},
 				},
 			},
@@ -148,43 +142,7 @@ func TestEnsureAssignmentsToActiveBrokersX(t *testing.T) {
 					{
 						LeaderBroker: "localhost:100",
 						Partition:    &mq_pb.Partition{},
-						FollowerBrokers: []string{
-							"localhost:200",
-						},
-					},
-				},
-			},
-			hasChanges: true,
-		},
-		{
-			name: "test missing two followers",
-			args: args{
-				activeBrokers: activeBrokers,
-				followerCount: 3,
-				assignments: []*mq_pb.BrokerPartitionAssignment{
-					{
-						LeaderBroker: "localhost:1",
-						Partition:    &mq_pb.Partition{},
-						FollowerBrokers: []string{
-							"localhost:2",
-						},
-					},
-				},
-			},
-			hasChanges: true,
-		},
-		{
-			name: "test missing some followers",
-			args: args{
-				activeBrokers: activeBrokers,
-				followerCount: 10,
-				assignments: []*mq_pb.BrokerPartitionAssignment{
-					{
-						LeaderBroker: "localhost:1",
-						Partition:    &mq_pb.Partition{},
-						FollowerBrokers: []string{
-							"localhost:2",
-						},
+						FollowerBroker: "localhost:200",
 					},
 				},
 			},
@@ -199,9 +157,7 @@ func TestEnsureAssignmentsToActiveBrokersX(t *testing.T) {
 					{
 						LeaderBroker: "localhost:1",
 						Partition:    &mq_pb.Partition{},
-						FollowerBrokers: []string{
-							"localhost:2",
-						},
+						FollowerBroker: "localhost:2",
 					},
 				},
 			},
@@ -230,9 +186,7 @@ func TestEnsureAssignmentsToActiveBrokersX(t *testing.T) {
 					{
 						LeaderBroker: "localhost:1",
 						Partition:    &mq_pb.Partition{},
-						FollowerBrokers: []string{
-							"localhost:2",
-						},
+						FollowerBroker: "localhost:2",
 					},
 				},
 			},