Fix to display of success message