Success message bug fixed