Fix message listener