Fix up undotree a little bit