diff --git a/storage-sample/src/main/java/com/omh/android/storage/sample/presentation/file_viewer/FileViewerActivity.kt b/storage-sample/src/main/java/com/omh/android/storage/sample/presentation/file_viewer/FileViewerActivity.kt
index aef1c5d2..7b78c00b 100644
--- a/storage-sample/src/main/java/com/omh/android/storage/sample/presentation/file_viewer/FileViewerActivity.kt
+++ b/storage-sample/src/main/java/com/omh/android/storage/sample/presentation/file_viewer/FileViewerActivity.kt
@@ -68,7 +68,7 @@ class FileViewerActivity :
                 val fileName = if (viewModel.isUpload) {
                     DocumentFile.fromSingleUri(this, uri)?.name
                 } else {
-                    viewModel.selectedFile?.name
+                    viewModel.lastFileClicked?.name
                 } ?: FileViewerViewModel.DEFAULT_FILE_NAME
                 showUploadFileDialog(uri, fileName)
             }
@@ -200,7 +200,7 @@ class FileViewerActivity :
     }
 
     override fun onUpdateClicked(file: OmhFile) {
-        viewModel.selectedFile = file
+        viewModel.lastFileClicked = file
         viewModel.isUpload = false
         filePicker.launch(FileViewerViewModel.ANY_MIME_TYPE)
     }
diff --git a/storage-sample/src/main/java/com/omh/android/storage/sample/presentation/file_viewer/FileViewerViewModel.kt b/storage-sample/src/main/java/com/omh/android/storage/sample/presentation/file_viewer/FileViewerViewModel.kt
index 94913a1e..2f1a4ffe 100644
--- a/storage-sample/src/main/java/com/omh/android/storage/sample/presentation/file_viewer/FileViewerViewModel.kt
+++ b/storage-sample/src/main/java/com/omh/android/storage/sample/presentation/file_viewer/FileViewerViewModel.kt
@@ -38,12 +38,11 @@ class FileViewerViewModel @Inject constructor(
         const val DEFAULT_FILE_NAME = "Untitled"
     }
 
-    var selectedFile: OmhFile? = null
     var isUpload = false
     var isGridLayoutManager = true
     var createFileSelectedType: OmhFileType? = null
     private val parentIdStack = Stack<String>().apply { push(ID_ROOT) }
-    private var lastFileClicked: OmhFile? = null
+    var lastFileClicked: OmhFile? = null
 
     override fun getInitialState(): FileViewerViewState = FileViewerViewState.Initial
 
@@ -141,7 +140,7 @@ class FileViewerViewModel @Inject constructor(
     }
 
     private fun updateFile(event: FileViewerViewEvent.UpdateFile) {
-        val file = selectedFile ?: return
+        val file = lastFileClicked ?: return
 
         setState(FileViewerViewState.Loading)