Fix flash message showing