diff --git a/app/FakeModel.php b/app/FakeModel.php
index 58636eea..13c29ab8 100644
--- a/app/FakeModel.php
+++ b/app/FakeModel.php
@@ -2,9 +2,9 @@
 
 namespace App;
 
+use App\Services\Media\HasMediaTrait;
 use Illuminate\Database\Eloquent\Model;
 use Spatie\MediaLibrary\HasMedia\HasMedia;
-use Spatie\MediaLibrary\HasMedia\HasMediaTrait;
 
 class FakeModel extends Model implements HasMedia
 {
diff --git a/app/Http/Controllers/Media/StoreMedia.php b/app/Http/Controllers/Media/StoreMedia.php
index 9231ebf9..2c108cf9 100644
--- a/app/Http/Controllers/Media/StoreMedia.php
+++ b/app/Http/Controllers/Media/StoreMedia.php
@@ -17,7 +17,7 @@ class StoreMedia extends Controller
         ], [
             'required' => trans('validations.required'),
             'max' => trans('validations.max', ['number' => 255]),
-            'mimes' => trans('validations.mimes', ['types' => 'jpeg,jpg,bmp,png']),
+            'mimes' => trans('validations.mimes', ['types' => 'jpeg,jpg,bmp,png,pdf']),
         ]);
         $fakeModel = FakeModel::firstOrCreate(['batch_id' => $request->input('batch_id')]);
         $media = $fakeModel->addMedia($request->file('file'))->toMediaCollection('temp_medias');
diff --git a/app/Item.php b/app/Item.php
index d0c698be..f4068ea4 100644
--- a/app/Item.php
+++ b/app/Item.php
@@ -4,7 +4,7 @@ namespace App;
 
 use Illuminate\Database\Eloquent\Model;
 use Spatie\MediaLibrary\HasMedia\HasMedia;
-use Spatie\MediaLibrary\HasMedia\HasMediaTrait;
+use App\Services\Media\HasMediaTrait;
 use Spatie\MediaLibrary\Models\Media;
 
 class Item extends Model implements HasMedia
diff --git a/app/TestResult.php b/app/TestResult.php
index 1a205880..2a06bdaf 100644
--- a/app/TestResult.php
+++ b/app/TestResult.php
@@ -4,7 +4,7 @@ namespace App;
 
 use Illuminate\Database\Eloquent\Model;
 use Spatie\MediaLibrary\HasMedia\HasMedia;
-use Spatie\MediaLibrary\HasMedia\HasMediaTrait;
+use App\Services\Media\HasMediaTrait;
 use Spatie\MediaLibrary\Models\Media;
 
 class TestResult extends Model implements HasMedia