diff --git a/resources/cider-ui-tests/index.js b/resources/cider-ui-tests/index.js
index 58b37345..1557cd03 100644
--- a/resources/cider-ui-tests/index.js
+++ b/resources/cider-ui-tests/index.js
@@ -129,6 +129,12 @@ const app = new Vue({
             // Set profile name
             this.chrome.userinfo = await this.mkapi("personalSocialProfile", false, "")
 
+            // load cached library
+            if(localStorage.getItem("librarySongs") != null) {
+                this.library.songs.listing = JSON.parse(localStorage.getItem("librarySongs"))
+                this.library.songs.displayListing = this.library.songs.listing
+            }
+
             MusicKit.getInstance().videoContainerElement = document.getElementById("apple-music-video-player")
             
             this.mk.addEventListener(MusicKit.Events.playbackTimeDidChange, (a) => {
@@ -792,14 +798,19 @@ const app = new Vue({
         playMediaItemById(id, kind, isLibrary, raurl = "") {
             var truekind = (!kind.endsWith("s")) ? (kind + "s") : kind;
             console.log(id, truekind, isLibrary)
-            if (truekind == "radioStations") {
-                this.mk.setStationQueue({url: raurl}).then(function (queue) {
-                    MusicKit.getInstance().play()
-                });
-            } else {
-                this.mk.setQueue({[truekind]: [id]}).then(function (queue) {
-                    MusicKit.getInstance().play()
-                })
+            try {
+                if (truekind == "radioStations") {
+                    this.mk.setStationQueue({url: raurl}).then(function (queue) {
+                        MusicKit.getInstance().play()
+                    });
+                } else {
+                    this.mk.setQueue({[truekind]: [id]}).then(function (queue) {
+                        MusicKit.getInstance().play()
+                    })
+                }
+            }catch(err){
+                console.log(err)
+                this.playMediaItemById(id, kind, isLibrary, raurl)
             }
         },
         friendlyTypes(type) {
@@ -872,17 +883,12 @@ const app = new Vue({
                 self.search.results = results
             })
         },
-        isInLibrary() {
+        isInLibrary(id) {
+            let self = this
             // ugly code to check if current playback item is in library
             var found = this.library.songs.listing.filter((item)=>{
-                var playingNow = "";
-                if(this.mk.nowPlayingItem["attributes"]){
-                    if(this.mk.nowPlayingItem["attributes"]["playParams"] && this.mk.nowPlayingItem["attributes"]["playParams"]["catalogId"]){
-                        playingNow = this.mk.nowPlayingItem["attributes"]["playParams"]["catalogId"];
-                    }
-                }
                 if(item["attributes"]){
-                    if(item["attributes"]["playParams"] && item["attributes"]["playParams"]["catalogId"] == playingNow){
+                    if(item["attributes"]["playParams"] && (item["attributes"]["playParams"]["catalogId"] == id)){
                         return item;
                     }
                 }
diff --git a/resources/cider-ui-tests/style.css b/resources/cider-ui-tests/style.css
index f798839b..69d8c186 100644
--- a/resources/cider-ui-tests/style.css
+++ b/resources/cider-ui-tests/style.css
@@ -120,6 +120,7 @@ body[loading] {
     opacity: 0.70;
     transition: opacity .25s var(--appleEase);
     pointer-events: none;
+    transform: translateZ(0px);
 }
 
 
diff --git a/resources/cider-ui-tests/views/main.ejs b/resources/cider-ui-tests/views/main.ejs
index 51a70ba3..a4f87686 100644
--- a/resources/cider-ui-tests/views/main.ejs
+++ b/resources/cider-ui-tests/views/main.ejs
@@ -75,7 +75,11 @@
                                             :max="mk.currentPlaybackDuration" :value="playerLCD.playbackDuration">
                                     </div>
                                 </div>
-                                <div class="actions">❤️</div>
+                                <template v-if="mk.nowPlayingItem['attributes']['playParams']['id']">
+                                    <div class="actions" v-if="isInLibrary(mk.nowPlayingItem['attributes']['playParams']['id'])">❤️</div>
+                                    <div class="actions" v-else>🖤</div>
+                                </template>
+                                
                             </div>
                         </template>