diff --git a/composer.json b/composer.json
index 87ac70e..72af1d9 100644
--- a/composer.json
+++ b/composer.json
@@ -9,11 +9,11 @@
     "prefer-stable": true,
     "autoload": {
         "psr-4": {
-            "": "src/"
+            "": "src/",
+            "EE\\Model\\": "src/db/"
         },
         "files": [
             "cron-command.php",
-            "src/helper/hooks.php",
             "src/helper/utils.php"
         ]
     },
diff --git a/migrations/db/20201111164339_cron-command_update_cron_user.php b/migrations/db/20201111164339_cron-command_update_cron_user.php
index 2f34392..885bdb6 100644
--- a/migrations/db/20201111164339_cron-command_update_cron_user.php
+++ b/migrations/db/20201111164339_cron-command_update_cron_user.php
@@ -5,6 +5,7 @@
 use EE;
 use EE\Migration\Base;
 use EE\Model\Cron;
+use EE\Model\Site;
 
 class UpdateCronUser extends Base {