Fix message hook