tests - Added wait for the widget-message