Fix tree not compatible with cell widget