Error flash message fix