Я хочу, чтобы в моем архиве категорий товаров woocommerce отображалось не то изображение, которое использовалось для изображения товара

Question

Я использовал следующий код из другого вопроса stackoverflow:

add_filter( 'woocommerce_product_gallery_attachment_ids', 'so_31818331_filter_gallery', 10, 2 );
function so_31818331_filter_gallery( $ids, $product ){
    if (is_product()) {
        set_post_thumbnail( $post, $ids[0] );
        unset($ids[0]);
    }
    else {
        set_post_thumbnail( $post, end($ids));
    }
    return $ids;
}

Используется последнее изображение из галереи для миниатюры, однако оно также меняет изображение товара / рекомендуемое изображение на то же изображение, которое мне не нужно.