Correct flash message
Use correct flash class