diff --git a/src/Concerns/HasMediaLibrary.php b/src/Concerns/HasMediaLibrary.php
index 3cc8e10c..1597a84f 100644
--- a/src/Concerns/HasMediaLibrary.php
+++ b/src/Concerns/HasMediaLibrary.php
@@ -25,7 +25,7 @@ trait HasMediaLibrary {
      */
     protected function getMediaModel() : HasMedia
     {
-        $model = Flexible::getOriginModel() ?? $this->model;
+        $model = Flexible::getOriginModel() ?? $this->model->model ?: $this->model;
 
         while ($model instanceof Layout) {
           $model = $model->getMediaModel();