Minor bug for message count