diff --git a/spec/Entity/WishlistSpec.php b/spec/Entity/WishlistSpec.php
index 884ca23e..de55eca0 100644
--- a/spec/Entity/WishlistSpec.php
+++ b/spec/Entity/WishlistSpec.php
@@ -47,6 +47,8 @@ function it_adds_wishlist_product(WishlistProductInterface $wishlistProduct, Pro
     {
         $wishlistProduct->getProduct()->willReturn($product);
 
+        $wishlistProduct->setWishlist($this)->shouldBeCalled();
+
         $this->hasWishlistProduct($wishlistProduct)->shouldReturn(false);
 
         $this->addWishlistProduct($wishlistProduct);