diff --git a/Controller/ApiController.php b/Controller/ApiController.php
index a6ec253..2455d28 100755
--- a/Controller/ApiController.php
+++ b/Controller/ApiController.php
@@ -22,6 +22,7 @@ use Modules\ContractManagement\Models\ContractTypeL11n;
 use Modules\ContractManagement\Models\ContractTypeL11nMapper;
 use Modules\ContractManagement\Models\ContractTypeMapper;
 use Modules\ContractManagement\Models\NullContractType;
+use Modules\Media\Models\MediaMapper;
 use Modules\Media\Models\PathSettings;
 use Modules\Organization\Models\NullUnit;
 use phpOMS\Localization\ISO639x1Enum;
@@ -144,16 +145,29 @@ final class ApiController extends Controller
         }
 
         $uploaded = $this->app->moduleManager->get('Media')->uploadFiles(
-            $request->getDataList('names'),
-            $request->getDataList('filenames'),
-            $uploadedFiles,
-            $request->header->account,
-            __DIR__ . '/../../../Modules/Media/Files/Modules/ContractManagement/Contracts/' . ($request->getData('contract_title') ?? '0'),
-            '/Modules/ContractManagement/Contracts/' . ($request->getData('contract_title') ?? '0'),
-            $request->getData('type', 'int'),
+            names: $request->getDataList('names'),
+            fileNames: $request->getDataList('filenames'),
+            files: $uploadedFiles,
+            account: $request->header->account,
+            basePath: __DIR__ . '/../../../Modules/Media/Files/Modules/ContractManagement/Contracts/' . ($request->getData('contract_title') ?? '0'),
+            virtualPath: '/Modules/ContractManagement/Contracts/' . ($request->getData('contract_title') ?? '0'),
             pathSettings: PathSettings::FILE_PATH
         );
 
+        if ($request->hasData('type')) {
+            foreach ($uploaded as $file) {
+                $this->createModelRelation(
+                    $request->header->account,
+                    $file->getId(),
+                    $request->getData('type', 'int'),
+                    MediaMapper::class,
+                    'types',
+                    '',
+                    $request->getOrigin()
+                );
+            }
+        }
+
         $this->createModelRelation(
             $request->header->account,
             (int) $request->getData('contract'),
diff --git a/Models/ContractMapper.php b/Models/ContractMapper.php
index c5f848b..8b3c06e 100755
--- a/Models/ContractMapper.php
+++ b/Models/ContractMapper.php
@@ -82,7 +82,7 @@ final class ContractMapper extends DataMapperFactory
     /**
      * Has one relation.
      *
-     * @var array<string, array{mapper:string, external:string, by?:string, column?:string, conditional?:bool}>
+     * @var array<string, array{mapper:class-string, external:string, by?:string, column?:string, conditional?:bool}>
      * @since 1.0.0
      */
     public const OWNS_ONE = [
@@ -103,7 +103,7 @@ final class ContractMapper extends DataMapperFactory
     /**
      * Has many relation.
      *
-     * @var array<string, array{mapper:string, table:string, self?:?string, external?:?string, column?:string}>
+     * @var array<string, array{mapper:class-string, table:string, self?:?string, external?:?string, column?:string}>
      * @since 1.0.0
      */
     public const HAS_MANY = [
diff --git a/Models/ContractTypeMapper.php b/Models/ContractTypeMapper.php
index 4d2793e..3a67519 100755
--- a/Models/ContractTypeMapper.php
+++ b/Models/ContractTypeMapper.php
@@ -39,7 +39,7 @@ final class ContractTypeMapper extends DataMapperFactory
     /**
      * Has many relation.
      *
-     * @var array<string, array{mapper:string, table:string, self?:?string, external?:?string, column?:string}>
+     * @var array<string, array{mapper:class-string, table:string, self?:?string, external?:?string, column?:string}>
      * @since 1.0.0
      */
     public const HAS_MANY = [