Fix updateTree