correct typehint

CHANGELOG.md

@@ -2,6 +2,9 @@
 
 All Notable changes to `laravel-medialibrary` will be documented in this file
 
+##2.1.1
+- Bugfix: correct typehint in HasMediaTrait
+
 ##2.1.0
 - Added some convenience methods for some frequent used manipulations
 

src/HasMediaTrait.php

@@ -169,10 +169,10 @@ public function updateMedia(array $newMediaArray, $collectionName = 'default')
     protected function removeMediaItemsNotPresentInArray(array $newMediaArray, $collectionName = 'default')
     {
         $this->getMedia($collectionName, [])
-            ->filter(function (HasMedia $currentMediaItem) use ($newMediaArray) {
+            ->filter(function (Media $currentMediaItem) use ($newMediaArray) {
                 return !in_array($currentMediaItem->id, collect($newMediaArray)->lists('id')->toArray());
             })
-            ->map(function (HasMedia $media) {
+            ->map(function (Media $media) {
                 $media->delete();
             });
     }