diff --git a/app/Notifications/Channels/SiteMessage.php b/app/Notifications/Channels/SiteMessage.php
index 27844bb7..153f8fb0 100644
--- a/app/Notifications/Channels/SiteMessage.php
+++ b/app/Notifications/Channels/SiteMessage.php
@@ -4,12 +4,15 @@
 
 namespace App\Notifications\Channels;
 
-use App\Casts\NotificationTypeEnum;
+use App\Models\Enums\NotificationTypeEnum;
 use Carbon\Carbon;
 use Illuminate\Notifications\Messages\DatabaseMessage;
+use Illuminate\Support\Traits\Conditionable;
 
 class SiteMessage extends DatabaseMessage
 {
+    use Conditionable;
+
     /**
      * @param array $data
      */