Flash message fixes