Fixed balloc2 to update tree in the same way as balloc.