diff --git a/Admin/Uninstall.php b/Admin/Uninstall.php
index 81eb299..a405aa3 100644
--- a/Admin/Uninstall.php
+++ b/Admin/Uninstall.php
@@ -17,6 +17,7 @@ namespace Modules\RiskManagement\Admin;
 use phpOMS\DataStorage\Database\DatabasePool;
 use phpOMS\DataStorage\Database\Schema\Builder;
 use phpOMS\Module\UninstallAbstract;
+use phpOMS\Module\InfoManager;
 
 /**
  * Navigation class.
diff --git a/Admin/Update.php b/Admin/Update.php
index 35ac68e..7486dd8 100644
--- a/Admin/Update.php
+++ b/Admin/Update.php
@@ -17,6 +17,8 @@ namespace Modules\RiskManagement\Admin;
 use phpOMS\DataStorage\Database\DatabasePool;
 use phpOMS\Module\UpdateAbstract;
 use phpOMS\System\File\Directory;
+use phpOMS\Module\InfoManager;
+
 
 /**
  * Navigation class.
@@ -33,7 +35,7 @@ class Update extends UpdateAbstract
     /**
      * {@inheritdoc}
      */
-    public static function update(DatabasePool $dbPool, array $info)
+    public static function update(DatabasePool $dbPool, InfoManager $info)
     {
         Directory::deletePath(__DIR__ . '/Update');
         mkdir('Update');