Fixed flash message bug - now disappears when resetting