Fixed bug related to tree widget