diff --git a/Album/album_edit.php b/Album/album_edit.php
index 35d1b93..fbad8f7 100644
--- a/Album/album_edit.php
+++ b/Album/album_edit.php
@@ -26,7 +26,7 @@ for ($i=0; $i<count($ApiAlbum['data']); $i++) {
     }
 }
 
-if ($ApiAlbum['data'][$i]['id'] == null) {
+if ($Album['id'] == null) {
     header('location: ./album.php');
 }