add the widget class back in