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 1fcf3f5f..78521f31 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
@@ -150,6 +150,7 @@ class FileViewerViewModel @Inject constructor(
     }
 
     private fun updateFileEvent(event: FileViewerViewEvent.UpdateFile) {
+        setState(FileViewerViewState.Loading)
         val fileId = lastFileClicked?.id
 
         if (fileId.isNullOrBlank()) {