Fix artwork_url for Spotify albums

1 files changed, 5 insertions, 4 deletions

diff --git a/code/utils/custom_sources.py b/code/utils/custom_sources.py
index 240947d..5cf2295 100644
--- a/code/utils/custom_sources.py
+++ b/code/utils/custom_sources.py
@@ -98,14 +98,15 @@ class SpotifySource(Source):
         )
 
     async def load_album(self, user, metadata):
+        try:
+            artwork_url = metadata["images"][0]["url"]
+        except IndexError:
+            artwork_url = None
+
         tracks = []
         for track in metadata["tracks"][
             "items"
         ]:  # Loop through each track in the album.
-            try:
-                artwork_url = track["album"]["images"][0]["url"]
-            except IndexError:
-                artwork_url = None
             tracks.append(
                 CustomAudioTrack(
                     {  # Create an instance of our CustomAudioTrack.