Fix to testing now that tree is different widget.