Fixed error in computing the set A (frozen bits)