diff --git a/GalleryImage.php b/GalleryImage.php
index 76db224..fccc553 100644
--- a/GalleryImage.php
+++ b/GalleryImage.php
@@ -2,7 +2,9 @@
 
 namespace zxbodya\yii2\galleryManager;
 
-class GalleryImage
+use yii\base\Model;
+
+class GalleryImage extends Model
 {
     public $name;
     public $description;
@@ -37,4 +39,11 @@ public function getUrl($version)
     {
         return $this->galleryBehavior->getUrl($this->id, $version);
     }
+
+    /**
+     * @return GalleryBehavior
+     */
+    public function getGalleryBehavior(){
+        return $this->galleryBehavior;
+    }
 }