Use wait_for_widget in widgets and string widget tests