Updated WidgetControllerTest