Fixing flash message bug.