diff --git a/weed/mq/pub_balancer/lookup.go b/weed/mq/pub_balancer/lookup.go
index 209261764..249aaaa1f 100644
--- a/weed/mq/pub_balancer/lookup.go
+++ b/weed/mq/pub_balancer/lookup.go
@@ -50,7 +50,7 @@ func (balancer *Balancer) LookupOrAllocateTopicPartitions(topic *mq_pb.Topic, pu
 	if balancer.Brokers.IsEmpty() {
 		return nil, ErrNoBroker
 	}
-	assignments = allocateTopicPartitions(balancer.Brokers, partitionCount)
+	assignments = AllocateTopicPartitions(balancer.Brokers, partitionCount)
 	balancer.OnPartitionChange(topic, assignments)
 	return
 }