diff --git a/contrib/preflate/support/huffman_helper.cpp b/contrib/preflate/support/huffman_helper.cpp index b3638f5..542979b 100644 --- a/contrib/preflate/support/huffman_helper.cpp +++ b/contrib/preflate/support/huffman_helper.cpp @@ -70,5 +70,6 @@ bool HuffmanHelper::countSymbols( } // Check that we don't have holes - return nextCode[maxLength] + blCount[maxLength] == (unsigned)(1 << (maxLength - 1)); + unsigned codeCheck = nextCode[maxLength] + blCount[maxLength]; + return codeCheck == (unsigned)(1 << (maxLength - 1)) || codeCheck == 1 && maxLength == 2; }