Fix message chunk state check