fix balance tree bug