Fix tests + fix flash messages on views