Change default product image to be the first one at checkout page

Hello,

I have e-com.express theme in use but there is a small problem. I have set the default image by drag and drop editor. However, when a visitor clicks the default product image he goes to the checkout page where the default image is not visible as the first image. There is a variant image instead.

Tried to fix this in the Sections and editing product-template.liquid and changing the first line:

{% assign current_variant = product.selected_or_first_available_variant %}

to this:
{% assign featured_media = product.featured_media %}

This fix worked and changed the default images to be the first ones at every checkout page.

Unfortunately, after this all the checkout pages lost Buy Now and Add To Cart buttons.

What to do to make this work?