diff --git a/backend/src/album/middleware/album.middleware.ts b/backend/src/album/middleware/album.middleware.ts
new file mode 100644
index 0000000..e69de29
diff --git a/frontend/src/app/album/[id]/page.tsx b/frontend/src/app/album/[id]/page.tsx
index 40fb632..d405ef0 100644
--- a/frontend/src/app/album/[id]/page.tsx
+++ b/frontend/src/app/album/[id]/page.tsx
@@ -76,7 +76,7 @@ export default function Page() {
         setFormModalTitle("Edit Song");
         setFormModalButtonLabel("Save");
         setFormSongId(songId)
-        setFormAlbumId(data.album.id);
+        setFormAlbumId(albumId);
         setFormSongTitle(data.title);
         setFormSongDuration(data.duration);
         setFormSongTrackNumber(data.trackNumber);