diff --git a/src/Resources/config/doctrine/ShippingExport.orm.yml b/src/Resources/config/doctrine/ShippingExport.orm.yml
index c2a7d62..44846b1 100644
--- a/src/Resources/config/doctrine/ShippingExport.orm.yml
+++ b/src/Resources/config/doctrine/ShippingExport.orm.yml
@@ -23,7 +23,7 @@ BitBag\SyliusShippingExportPlugin\Entity\ShippingExport:
             nullable: true
     oneToOne:
         shipment:
-            targetEntity: Sylius\Component\Core\Model\Shipment
+            targetEntity: Sylius\Component\Shipping\Model\ShipmentInterface
             joinColumn:
                 name: shipment_id
                 referencedColumnName: id