I'm not seeing your point immediately but I have only begun trying to page this into brainRAM (bRAM?). My immediate thought is that I'm missing the difference between the rebalance/count_by_node and the current shard map. I'll read this bit closer tomorrow.