Fix balancer test