diff --git a/app/src/main/java/org/lineageos/glimpse/viewmodels/MediaViewModel.kt b/app/src/main/java/org/lineageos/glimpse/viewmodels/MediaViewModel.kt
index d7bd8f9..215c53b 100644
--- a/app/src/main/java/org/lineageos/glimpse/viewmodels/MediaViewModel.kt
+++ b/app/src/main/java/org/lineageos/glimpse/viewmodels/MediaViewModel.kt
@@ -5,11 +5,9 @@
 
 package org.lineageos.glimpse.viewmodels
 
-import androidx.lifecycle.SavedStateHandle
 import androidx.lifecycle.ViewModel
 import androidx.lifecycle.ViewModelProvider
 import androidx.lifecycle.ViewModelProvider.AndroidViewModelFactory.Companion.APPLICATION_KEY
-import androidx.lifecycle.createSavedStateHandle
 import androidx.lifecycle.viewmodel.initializer
 import androidx.lifecycle.viewmodel.viewModelFactory
 import kotlinx.coroutines.ExperimentalCoroutinesApi
@@ -20,7 +18,7 @@
 import org.lineageos.glimpse.utils.MediaStoreBuckets
 
 open class MediaViewModel(
-    private val savedStateHandle: SavedStateHandle, private val mediaRepository: MediaRepository
+    private val mediaRepository: MediaRepository
 ) : ViewModel() {
     val media = mediaRepository.media(MediaStoreBuckets.MEDIA_STORE_BUCKET_REELS.id)
     val albums = mediaRepository.albums()
@@ -37,7 +35,6 @@
         val Factory: ViewModelProvider.Factory = viewModelFactory {
             initializer {
                 MediaViewModel(
-                    savedStateHandle = createSavedStateHandle(),
                     mediaRepository = (this[APPLICATION_KEY] as GlimpseApplication).mediaRepository,
                 )
             }
diff --git a/app/src/main/java/org/lineageos/glimpse/viewmodels/MediaViewerViewModel.kt b/app/src/main/java/org/lineageos/glimpse/viewmodels/MediaViewerViewModel.kt
index 867c82c..30b053e 100644
--- a/app/src/main/java/org/lineageos/glimpse/viewmodels/MediaViewerViewModel.kt
+++ b/app/src/main/java/org/lineageos/glimpse/viewmodels/MediaViewerViewModel.kt
@@ -19,7 +19,7 @@
 class MediaViewerViewModel(
     savedStateHandle: SavedStateHandle,
     mediaRepository: MediaRepository,
-) : MediaViewModel(savedStateHandle, mediaRepository) {
+) : MediaViewModel(mediaRepository) {
     private val mediaPositionInternal = savedStateHandle.getLiveData<Int>(MEDIA_POSITION_KEY)
     val mediaPositionLiveData: LiveData<Int> = mediaPositionInternal
     var mediaPosition: Int