Fix flashmessage tests