diff --git a/composeApp/src/androidMain/kotlin/it/fast4x/rimusic/ui/screens/album/AlbumScreen.kt b/composeApp/src/androidMain/kotlin/it/fast4x/rimusic/ui/screens/album/AlbumScreen.kt
index fa494ccd0c..1608c76d87 100644
--- a/composeApp/src/androidMain/kotlin/it/fast4x/rimusic/ui/screens/album/AlbumScreen.kt
+++ b/composeApp/src/androidMain/kotlin/it/fast4x/rimusic/ui/screens/album/AlbumScreen.kt
@@ -156,11 +156,11 @@ fun AlbumScreen(
                                     ),
                                     currentAlbumPage
                                         .songs.distinct()
-                                        .map(Innertube.SongItem::asSong)
+                                        .map(Innertube.SongItem::asMediaItem)
                                         .onEach(Database::insert)
-                                        .mapIndexed { position, song ->
+                                        .mapIndexed { position, mediaItem ->
                                             SongAlbumMap(
-                                                songId = song.id,
+                                                songId = mediaItem.mediaId,
                                                 albumId = browseId,
                                                 position = position
                                             )