Fix collapsing tree node cause cellwidget missing