Adapt two tests which failed after the flash message on toggle was removed.