diff --git a/src/Offer/Offer.php b/src/Offer/Offer.php
index <HASH>..<HASH> 100644
--- a/src/Offer/Offer.php
+++ b/src/Offer/Offer.php
@@ -659,7 +659,7 @@ abstract class Offer extends EventSourcedAggregateRoot implements LabelAwareAggr
      */
     public function selectMainImage(Image $image)
     {
-        if (!$this->images->contains($image)) {
+        if (!$this->images->findImageByUUID($image->getMediaObjectId())) {
             throw new \InvalidArgumentException('You can not select a random image to be main, it has to be added to the item.');
         }
III-<I> Search image based on uuid.