Fixed bug causing success message to not display