diff --git a/src/Entity/ShipmentTrait.php b/src/Entity/ShipmentTrait.php
index 6ecdd7e..c70882e 100644
--- a/src/Entity/ShipmentTrait.php
+++ b/src/Entity/ShipmentTrait.php
@@ -18,7 +18,7 @@
 trait ShipmentTrait
 {
     /**
-     * @ORM\OneToOne(targetEntity="MonsieurBiz\SyliusShippingSlotPlugin\Entity\Slot", mappedBy="shipment", fetch="EAGER")
+     * @ORM\OneToOne(targetEntity="MonsieurBiz\SyliusShippingSlotPlugin\Entity\SlotInterface", mappedBy="shipment", fetch="EAGER")
      */
     private ?SlotInterface $slot = null;