Here is a simple fix. This is straightforward change. I have verified it with TestBalancer.java, which is having a testcase for pinned blocks. So, even after return the current behavior should not change and it should work as is.