Variation color change does not change archive product pricture

Hello,

I have had on the archive page when the user selected some color from variations, then the main image would change into that color. That functionality is currently not working, can you help?

Best regards,