Image/ Color Swatches for Webflow e-commerce

Hey guys,

I wanted to ask if there’s a work-around to be able to add the image variant to the variant buttons in webflow e-commerce. At the moment you can only have buttons with text, which is not enough in some use-cases, for example when building a clothing store. Every store either has color swatches or image swatches. Is there a way to dynamically pull the color variant image to the button?

I’ll attach a photo as well.

Thanks!
Screenshot 2021-06-14 at 19.00.06