diff --git a/src/Livewire/CommentsComponent.php b/src/Livewire/CommentsComponent.php
index 021999d..88037ef 100644
--- a/src/Livewire/CommentsComponent.php
+++ b/src/Livewire/CommentsComponent.php
@@ -10,7 +10,6 @@
 use Illuminate\Contracts\View\View;
 use Illuminate\Database\Eloquent\Model;
 use Livewire\Component;
-use Parallax\FilamentComments\Models\FilamentComment;
 
 class CommentsComponent extends Component implements HasForms
 {
@@ -79,7 +78,7 @@ public function create(): void
 
     public function delete(int $id): void
     {
-        $comment = FilamentComment::find($id);
+        $comment = app(config('filament-comments.comment_model'))::find($id);
 
         if (!$comment) {
             return;