diff --git a/src/Core/Models/Model.php b/src/Core/Models/Model.php
index fa6a7e7a7..0aae45949 100755
--- a/src/Core/Models/Model.php
+++ b/src/Core/Models/Model.php
@@ -5,10 +5,12 @@ use Illuminate\Database\Eloquent\Model as Eloquent;
 use Flarum\Core\Exceptions\ValidationFailureException;
 use Flarum\Core\Exceptions\PermissionDeniedException;
 use Flarum\Core\Support\EventGenerator;
+use Flarum\Core\Support\MappedMorphToTrait;
 
 class Model extends Eloquent
 {
     use EventGenerator;
+    use MappedMorphToTrait;
 
     /**
      * Disable timestamps.
diff --git a/src/Core/Models/Notification.php b/src/Core/Models/Notification.php
index c22ce49e5..1aab33e72 100644
--- a/src/Core/Models/Notification.php
+++ b/src/Core/Models/Notification.php
@@ -1,11 +1,7 @@
 <?php namespace Flarum\Core\Models;
 
-use Flarum\Core\Support\MappedMorphToTrait;
-
 class Notification extends Model
 {
-    use MappedMorphToTrait;
-
     /**
      * The table associated with the model.
      *