Fix @covers for BalancerTest