Update to use generated widget.