diff --git a/src/InteractsWithMedia.php b/src/InteractsWithMedia.php
index a811c9f8f..4420ed751 100644
--- a/src/InteractsWithMedia.php
+++ b/src/InteractsWithMedia.php
@@ -58,7 +58,7 @@ public static function bootInteractsWithMedia()
 
     public function media(): MorphMany
     {
-        return $this->morphMany(config('media-library.media_model'), 'model');
+        return $this->morphMany($this->getMediaModel(), 'model');
     }
 
     /**
@@ -263,6 +263,11 @@ public function getMediaRepository(): MediaRepository
         return app(MediaRepository::class);
     }
 
+    public function getMediaModel(): string
+    {
+        return config('media-library.media_model');
+    }
+
     public function getFirstMedia(string $collectionName = 'default', $filters = []): ?Media
     {
         $media = $this->getMedia($collectionName, $filters);
@@ -378,7 +383,7 @@ public function updateMedia(array $newMediaArray, string $collectionName = 'defa
     {
         $this->removeMediaItemsNotPresentInArray($newMediaArray, $collectionName);
 
-        $mediaClass = config('media-library.media_model');
+        $mediaClass = $this->getMediaModel();
         $mediaInstance = new $mediaClass();
         $keyName = $mediaInstance->getKeyName();