diff --git a/app/Models/Media.php b/app/Models/Media.php
index 45e2e3c..e57db23 100644
--- a/app/Models/Media.php
+++ b/app/Models/Media.php
@@ -6,10 +6,10 @@
 use App\Traits\HasRatings;
 use App\Traits\HasComments;
 use App\Traits\HasSaves;
-
+use App\Traits\HasLikes;
 class Media extends Model
 {
-     use HasRatings, HasComments,HasSaves;
+     use HasRatings, HasComments,HasSaves,HasLikes;
   protected $fillable = [
         'user_id',
         'image_id',
@@ -35,7 +35,9 @@ class Media extends Model
         'is_saved', 
         'saved_count',
         'is_liked',   
-        'likes_count' 
+        'likes_count',
+'is_liked',
+        'likes_count', 
     ];
     public function image()
     {