fixing the last merge issue with TreeNode