Картинка выбранного варианта товара в WooCommerce

При создании аттрибутов у вариатвного товара, для каждого варианта установлены соответствующие изображения. Также установлена общая картинка товара в Product Image.
При выборе вариантов, картинки меняются, как и задумано. Но, если добавить товар в корзину с каким то выбранным вариантом, в корзине получаем правильную ссылку на товар (с вариантом) и при переходе аттрибут устанавливается и я получаю страницу товара, с выбранным вариантом, но показывается общая картинка товара, а не те которые назначены для конкретного варианта.
Подскажите пожалуйста, в какую сторону копать или как исправить.
Спасибо.