Fix flash message bug