diff --git a/app/src/main/java/awais/instagrabber/fragments/main/viewmodels/BasePostViewModel.java b/app/src/main/java/awais/instagrabber/fragments/main/viewmodels/BasePostViewModel.java
index 4dba6f4a..8aecc190 100644
--- a/app/src/main/java/awais/instagrabber/fragments/main/viewmodels/BasePostViewModel.java
+++ b/app/src/main/java/awais/instagrabber/fragments/main/viewmodels/BasePostViewModel.java
@@ -1,13 +1,9 @@
 package awais.instagrabber.fragments.main.viewmodels;
 
-import androidx.lifecycle.MutableLiveData;
 import androidx.lifecycle.ViewModel;
 
-import java.util.List;
-
 import awais.instagrabber.models.BasePostModel;
 
-public class BasePostViewModel extends ViewModel {
-    // this file was somehow identical to DiscoverItemViewModel
-    // pretty sure you didn't meant to do that
+public class BasePostViewModel<T extends BasePostModel> extends ViewModel {
+
 }
\ No newline at end of file
diff --git a/app/src/main/java/awais/instagrabber/fragments/main/viewmodels/ViewerPostViewModel.java b/app/src/main/java/awais/instagrabber/fragments/main/viewmodels/ViewerPostViewModel.java
index de935620..04a63365 100644
--- a/app/src/main/java/awais/instagrabber/fragments/main/viewmodels/ViewerPostViewModel.java
+++ b/app/src/main/java/awais/instagrabber/fragments/main/viewmodels/ViewerPostViewModel.java
@@ -1,15 +1,16 @@
 package awais.instagrabber.fragments.main.viewmodels;
 
 import androidx.lifecycle.MutableLiveData;
+import androidx.lifecycle.ViewModel;
 
 import java.util.List;
 
-import awais.instagrabber.models.ViewerPostModel;
+import awais.instagrabber.models.ViewerPostModelWrapper;
 
-public class ViewerPostViewModel extends BasePostViewModel<ViewerPostModel> {
-    private MutableLiveData<List<ViewerPostModel>> list;
+public class ViewerPostViewModel extends ViewModel {
+    private MutableLiveData<List<ViewerPostModelWrapper>> list;
 
-    public MutableLiveData<List<ViewerPostModel>> getList() {
+    public MutableLiveData<List<ViewerPostModelWrapper>> getList() {
         if (list == null) {
             list = new MutableLiveData<>();
         }