Fix the message count error