Minor bug fix
Update the tree