Fix message display error