diff --git a/src/Mapper/DocumentApiToEntityMapper.php b/src/Mapper/DocumentApiToEntityMapper.php
index b788801..cc3859e 100644
--- a/src/Mapper/DocumentApiToEntityMapper.php
+++ b/src/Mapper/DocumentApiToEntityMapper.php
@@ -57,14 +57,14 @@ class DocumentApiToEntityMapper implements MapperInterface
 
         $entity->setName($dto->name);
         $entity->setDescription($dto->description);
-        if ($dto->partner !== null) {
+        if ($dto->partnerIri !== null) {
             $entity->setPartner(
                 $this->microMapper->map($dto->partnerIri, Partner::class, [
                     MicroMapperInterface::MAX_DEPTH => 1,
                 ])
             );
         }
-        if ($dto->product !== null) {
+        if ($dto->productIri !== null) {
             $entity->setProduct(
                 $this->microMapper->map($dto->productIri, Product::class, [
                     MicroMapperInterface::MAX_DEPTH => 1,