Fixed tree edit widget bug