diff --git a/weed/mq/balancer/allocate_test.go b/weed/mq/balancer/allocate_test.go
index cb87d4ec3..f5f9d49e3 100644
--- a/weed/mq/balancer/allocate_test.go
+++ b/weed/mq/balancer/allocate_test.go
@@ -43,7 +43,7 @@ func Test_allocateOneBroker(t *testing.T) {
 
 type args struct {
 	brokers        cmap.ConcurrentMap[string, *BrokerStats]
-	partitionCount int
+	partitionCount int32
 }
 
 func testThem(t *testing.T, tests []struct {