diff --git a/src/FilamentAlertsServiceProvider.php b/src/FilamentAlertsServiceProvider.php
index a85773e..0f08ec9 100644
--- a/src/FilamentAlertsServiceProvider.php
+++ b/src/FilamentAlertsServiceProvider.php
@@ -65,6 +65,7 @@ public function register(): void
                 modelId: $user->id,
                 image: $data['image'] ?? null,
                 data: $data,
+                notification: $this
             );
 
             return $this;
diff --git a/src/Services/Drivers/DatabaseDriver.php b/src/Services/Drivers/DatabaseDriver.php
index 794f342..dbcce87 100644
--- a/src/Services/Drivers/DatabaseDriver.php
+++ b/src/Services/Drivers/DatabaseDriver.php
@@ -2,6 +2,7 @@
 
 namespace TomatoPHP\FilamentAlerts\Services\Drivers;
 
+use Filament\Notifications\Notification;
 use TomatoPHP\FilamentAlerts\Jobs\NotifyDatabaseJob;
 
 class DatabaseDriver extends Driver
diff --git a/src/Services/Drivers/EmailDriver.php b/src/Services/Drivers/EmailDriver.php
index 94e7eb1..affb226 100644
--- a/src/Services/Drivers/EmailDriver.php
+++ b/src/Services/Drivers/EmailDriver.php
@@ -2,6 +2,7 @@
 
 namespace TomatoPHP\FilamentAlerts\Services\Drivers;
 
+use Filament\Notifications\Notification;
 use TomatoPHP\FilamentAlerts\Jobs\NotifyEmailJob;
 
 class EmailDriver extends Driver