Fix message count error