Fix message bug