diff --git a/Models/Collection.php b/Models/Collection.php
index b175bd5..9bb375e 100644
--- a/Models/Collection.php
+++ b/Models/Collection.php
@@ -69,14 +69,14 @@ class Collection extends Media
     /**
      * Set sources.
      *
-     * @param int $source Source
+     * @param int|Media $source Source
      *
      * @return void
      *
      * @since  1.0.0
      * @author Dennis Eichhorn <d.eichhorn@oms.com>
      */
-    public function addSource(int $source)
+    public function addSource($source)
     {
         $this->sources[] = $source;
     }
diff --git a/Models/MediaMapper.php b/Models/MediaMapper.php
index bde82fd..7b7ef33 100644
--- a/Models/MediaMapper.php
+++ b/Models/MediaMapper.php
@@ -82,25 +82,32 @@ class MediaMapper extends DataMapperAbstract
     {
         try {
             $objId = parent::create($obj, $relations);
+
+            if($objId === null || !is_scalar($objId)) {
+                return $objId;
+            }
+
             $query = new Builder(self::$db);
             $query->prefix(self::$db->getPrefix())
-                  ->insert(
-                      'account_permission_account',
-                      'account_permission_from',
-                      'account_permission_for',
-                      'account_permission_id1',
-                      'account_permission_id2',
-                      'account_permission_r',
-                      'account_permission_w',
-                      'account_permission_m',
-                      'account_permission_d',
-                      'account_permission_p'
-                  )
-                  ->into('account_permission')
-                  ->values($obj->getCreatedBy(), 'media', 'media', 1, $objId, 1, 1, 1, 1, 1);
+                ->insert(
+                    'account_permission_account',
+                    'account_permission_from',
+                    'account_permission_for',
+                    'account_permission_id1',
+                    'account_permission_id2',
+                    'account_permission_r',
+                    'account_permission_w',
+                    'account_permission_m',
+                    'account_permission_d',
+                    'account_permission_p'
+                )
+                ->into('account_permission')
+                ->values($obj->getCreatedBy(), 'media', 'media', 1, $objId, 1, 1, 1, 1, 1);
 
             self::$db->con->prepare($query->toSql())->execute();
         } catch (\Exception $e) {
+            var_dump($e->getMessage());
+
             return false;
         }