Image gallery problem

My image gallery in product page is working fine, but when i change size for a product, gallery is messed up.

masina-shop.co.rs/index.php/product/show/261-cocaine

I have fixed thumb display, but now i have another problem.

I used code from this thread www.hikashop.com/en/support/forum/4-how-...n-product-image.html to hide first thumb and replace it with second thumb. That is working fine, but when i change variant, the first thumb comes back again.

I also tried this code but that doesn't help.

For the variants, you need to add the line:
$variant->images = array_shift($variant->images);

after the line:
foreach($this->element->variants as $variant){

Now i started adding some variants to my products, but variants wont update. I found that the problem is modifications that i made in the show file.

This is the code that is causing problem:
array_shift($variant->images);

When i delete this line variations are displaying fine.

Is there a way to fix this.

Thanks