I think so. I kept the original semantic.... Also, if you return true in both invocations, you will invoke ```endRebalance()``` twice (or more) and I don't think that it is safe...