Changed container dimensions after update

Hi,
After the update the variation images doesn’t show the same way anymore. How do I get i back to the correct way? If you look at the attached image the container now is rectangular but it was square before. That makes it change row so that it’s now only two variations on each row aginst three before = no good. How do I solve it?