Fix message handling