How to show corresponding product colour image after doing filtering?

[Boyi]

Context:

• We are using the Dawn theme and Shopify's search and Discovery app to do filters 
• For filtering, we allow the user to filter based on colours

Challenge:

• Currently, when the user selects a colour from the filter, all the product card's first image change to the featured image for that variant

        

• However, on hover, the product's second image is shown and not the scene image that corresponds to the filtered colour

         

Question:

• Any way to fix this?
• Link for consideration: https://cozylant.com/collections/sofas?filter.v.price.gte=&filter.v.price.lte=&filter.v.option.colou...

Thanks!