Fixed some logic

Setono · Dec 3, 2024 · f92110f · f92110f
1 parent fc05b68
commit f92110f
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/Model/WishlistItem.php b/src/Model/WishlistItem.php
@@ -6,6 +6,7 @@
 
 use Sylius\Component\Core\Model\ProductInterface;
 use Sylius\Component\Core\Model\ProductVariantInterface;
+use Webmozart\Assert\Assert;
 
 class WishlistItem implements WishlistItemInterface
 {
@@ -55,9 +56,9 @@ public function setVariant(?ProductVariantInterface $variant): void
 
         if (null !== $variant) {
             $product = $variant->getProduct();
-            if ($product instanceof ProductInterface) {
-                $this->product = $product;
-            }
+            Assert::isInstanceOf($product, ProductInterface::class);
+
+            $this->product = $product;
         }
     }