Add one more rebalance-related test