From 6860ef32846f0050db3d87c8d899ff999b8350d3 Mon Sep 17 00:00:00 2001
From: freek <freek@spatie.be>
Date: Mon, 13 Jul 2015 15:14:33 +0200
Subject: [PATCH] correct typehint

---
 CHANGELOG.md          | 3 +++
 src/HasMediaTrait.php | 4 ++--
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index a446d13ba..4cba1d91b 100644
--- a/CHANGELOG.md
+++ b/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
 
diff --git a/src/HasMediaTrait.php b/src/HasMediaTrait.php
index 31b886a71..abd15b94c 100644
--- a/src/HasMediaTrait.php
+++ b/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();
             });
     }