For ENG-5464, fix testRebalanceStats