Fix the flash message