Fix message handler bug