swatch as a variant image shopify 2.0

{%- for variant in product.variants -%}
{%- if variant.title contains value -%}
{%- assign variantImgSrc = variant.image.src | img_url: '300x' -%}
{%- break -%}
{%- endif -%}
{%- endfor -%}
{% if option.name == 'color' or option.name == 'Color' or option.name == 'Colour' or option.name == 'colour' %}
<img src="{{ variantImgSrc }}">
{% else %}
{{ value }}
{% endif %}