diff --git a/sample/src/main/java/com/google/modernstorage/sample/mediastore/MediaStoreViewModel.kt b/sample/src/main/java/com/google/modernstorage/sample/mediastore/MediaStoreViewModel.kt
index fe3e21aa..3d7b5864 100644
--- a/sample/src/main/java/com/google/modernstorage/sample/mediastore/MediaStoreViewModel.kt
+++ b/sample/src/main/java/com/google/modernstorage/sample/mediastore/MediaStoreViewModel.kt
@@ -62,13 +62,13 @@ class MediaStoreViewModel(application: Application) : AndroidViewModel(applicati
                 MediaType.VIDEO -> {
                     extension = "mp4"
                     mimeType = "video/mp4"
-                    collection = MediaStore.Images.Media.getContentUri("external")
+                    collection = MediaStore.Video.Media.getContentUri("external")
                     directory = Environment.getExternalStoragePublicDirectory(Environment.DIRECTORY_MOVIES)
                 }
                 MediaType.AUDIO -> {
                     extension = "wav"
                     mimeType = "audio/x-wav"
-                    collection = MediaStore.Images.Media.getContentUri("external")
+                    collection = MediaStore.Audio.Media.getContentUri("external")
                     directory = Environment.getExternalStoragePublicDirectory(Environment.DIRECTORY_MUSIC)
                 }
             }