Polygamy is one thing, and the punishment is another. Islam never punishes just the woman, but always both the woman and the man.