diff --git a/resources/cider-ui-tests/index.js b/resources/cider-ui-tests/index.js
index cd671f11..6ade23a3 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) => {
@@ -844,17 +850,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/views/main.ejs b/resources/cider-ui-tests/views/main.ejs
index d1c01cfd..b9b0ba11 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>