Fix message display