diff --git a/phpstan-baseline.neon b/phpstan-baseline.neon
index f8b725c..1fe2dc2 100644
--- a/phpstan-baseline.neon
+++ b/phpstan-baseline.neon
@@ -16,3 +16,7 @@ parameters:
 			message: "#^Call to an undefined method Illuminate\\\\Database\\\\Eloquent\\\\Builder\\:\\:search\\(\\).$#"
 			count: 1
 			path: src/Resources/AttachmentResource.php
+		-
+			message: "#^Call to an undefined method Livewire\\\\Features\\\\SupportFileUploads\\\\TemporaryUploadedFile\\:\\:save\\(\\).$#"
+			count: 2
+			path: src/Filament/Actions/Traits/CanUploadAttachment.php
diff --git a/src/Formats/Format.php b/src/Formats/Format.php
index e3fd055..deebd70 100644
--- a/src/Formats/Format.php
+++ b/src/Formats/Format.php
@@ -7,6 +7,7 @@
 use Codedor\MediaLibrary\Models\Attachment;
 use Illuminate\Contracts\Support\Arrayable;
 use Illuminate\Support\Str;
+use Spatie\Image\Drivers\ImageDriver;
 
 abstract class Format implements Arrayable
 {
@@ -18,7 +19,7 @@ abstract class Format implements Arrayable
 
     protected string $description;
 
-    abstract public function definition(): Manipulations;
+    abstract public function definition(): Manipulations|ImageDriver;
 
     abstract public function registerModelsForFormatter(): void;
 
diff --git a/src/Formats/Thumbnail.php b/src/Formats/Thumbnail.php
index e5ddebb..74cada1 100644
--- a/src/Formats/Thumbnail.php
+++ b/src/Formats/Thumbnail.php
@@ -3,6 +3,7 @@
 namespace Codedor\MediaLibrary\Formats;
 
 use Codedor\MediaLibrary\Models\Attachment;
+use Spatie\Image\Drivers\ImageDriver;
 use Spatie\Image\Enums\Fit;
 
 class Thumbnail extends Format
@@ -13,7 +14,7 @@ class Thumbnail extends Format
 
     protected string $description = 'Used in the CMS to display low-res images';
 
-    public function definition(): Manipulations
+    public function definition(): Manipulations|ImageDriver
     {
         return $this->manipulations->fit(Fit::Crop, 350, 350);
     }
diff --git a/src/Mixins/UploadedFileMixin.php b/src/Mixins/UploadedFileMixin.php
index b56589c..5593d8b 100644
--- a/src/Mixins/UploadedFileMixin.php
+++ b/src/Mixins/UploadedFileMixin.php
@@ -11,9 +11,10 @@
 use Illuminate\Support\Facades\Storage;
 use Illuminate\Support\Facades\Validator;
 use Illuminate\Support\Str;
+use Livewire\Features\SupportFileUploads\TemporaryUploadedFile;
 
 /**
- * @mixin UploadedFile
+ * @mixin TemporaryUploadedFile
  */
 class UploadedFileMixin
 {