Changed the tests to match the flash message changes