Fix message tests