diff --git a/Models/ContractMapper.php b/Models/ContractMapper.php
index afb3fc0..d0cbd88 100755
--- a/Models/ContractMapper.php
+++ b/Models/ContractMapper.php
@@ -45,7 +45,7 @@ final class ContractMapper extends DataMapperFactory
     public const COLUMNS = [
         'contractmgmt_contract_id'                        => ['name' => 'contractmgmt_contract_id',          'type' => 'int',               'internal' => 'id'],
         'contractmgmt_contract_parent'                    => ['name' => 'contractmgmt_contract_parent',          'type' => 'int',               'internal' => 'parent'],
-        'contractmgmt_contract_template'                  => ['name' => 'contractmgmt_contract_template',          'type' => 'int',               'internal' => 'isTemplate'],
+        'contractmgmt_contract_template'                  => ['name' => 'contractmgmt_contract_template',          'type' => 'bool',               'internal' => 'isTemplate'],
         'contractmgmt_contract_title'                     => ['name' => 'contractmgmt_contract_title',       'type' => 'string',            'internal' => 'title', 'autocomplete' => true],
         'contractmgmt_contract_description'               => ['name' => 'contractmgmt_contract_description', 'type' => 'string',            'internal' => 'description'],
         'contractmgmt_contract_account'                   => ['name' => 'contractmgmt_contract_account',     'type' => 'int',               'internal' => 'account'],
diff --git a/tests/Models/ContractTest.php b/tests/Models/ContractTest.php
index 3890063..4006b75 100755
--- a/tests/Models/ContractTest.php
+++ b/tests/Models/ContractTest.php
@@ -43,16 +43,6 @@ final class ContractTest extends \PHPUnit\Framework\TestCase
         self::assertEquals([], $this->contract->files);
     }
 
-    /**
-     * @covers Modules\ContractManagement\Models\Contract
-     * @group module
-     */
-    public function testMediaInputOutput() : void
-    {
-        $this->contract->addFile(new Media());
-        self::assertCount(1, $this->contract->files);
-    }
-
     /**
      * @covers Modules\ContractManagement\Models\Contract
      * @group module
@@ -78,7 +68,7 @@ final class ContractTest extends \PHPUnit\Framework\TestCase
                 'warning'        => 2,
                 'responsible'    => null,
                 'costs'          => null,
-                'type'           => new ContractType(),
+                'type'           => null,
             ],
             $serialized
         );