The code in this repo is not consistent with https://medium.com/thinkspecial/handling-partition-rebalancing-3e4f3cb192b0. Am i missing something ?