Fix message mapping