diff --git a/Admin/Updater.php b/Admin/Updater.php
index f7e600b..87d3351 100644
--- a/Admin/Updater.php
+++ b/Admin/Updater.php
@@ -16,7 +16,7 @@ namespace Modules\Media\Admin;
 
 use phpOMS\DataStorage\Database\DatabasePool;
 use phpOMS\Module\UpdaterAbstract;
-use phpOMS\System\File\Directory;
+use phpOMS\System\File\Local\Directory;
 use phpOMS\Module\InfoManager;
 
 /**
diff --git a/Models/MediaMapper.php b/Models/MediaMapper.php
index 0be9c37..c310d02 100644
--- a/Models/MediaMapper.php
+++ b/Models/MediaMapper.php
@@ -26,7 +26,7 @@ class MediaMapper extends DataMapperAbstract
     /**
      * Columns.
      *
-     * @var array<string, array<string, string>>
+     * @var array<string, array<string, string|bool>>
      * @since 1.0.0
      */
     protected static $columns = [