Swatch image from variant image

Hi,

When I watched the demo video of the pro version of the swatches, I assumed it would take the image associated with the variant it applies to. Is there any way to make it take this image if it’s associated rather than going through all the swatches and updating the image for each attribute?