Fix an issue in TreeNode where subtries would be forced to balance