diff --git a/resources/cider-ui-tests/index.js b/resources/cider-ui-tests/index.js
index 195dc9a2..b4e53abf 100644
--- a/resources/cider-ui-tests/index.js
+++ b/resources/cider-ui-tests/index.js
@@ -346,11 +346,11 @@ const app = new Vue({
                 return await this.mkapi(method, library, term, params, params2, attempts + 1)
             }
         },
-        async getLibrarySongsFull() {
+        async getLibrarySongsFull(force = false) {
             let self = this
             let library = []
             let downloaded = null;
-            if (this.library.songs.downloadState == 2 || this.library.songs.downloadState == 1) {
+            if ((this.library.songs.downloadState == 2 || this.library.songs.downloadState == 1) && !force) {
                 return
             }
             if(localStorage.getItem("librarySongs") != null) {
@@ -855,9 +855,17 @@ const app = new Vue({
                 self.search.results = results
             })
         },
-        isInLibrary(id) {
+        isInLibrary(playParams) {
             let self = this
+            let id = ""
             // ugly code to check if current playback item is in library
+            if(playParams.catalogId) {
+                id = playParams.catalogId
+            }else if(playParams.id) {
+                id = playParams.id
+            }else if(playParams.isLibrary) {
+                return true
+            }
             var found = this.library.songs.listing.filter((item)=>{
                 if(item["attributes"]){
                     if(item["attributes"]["playParams"] && (item["attributes"]["playParams"]["catalogId"] == id)){
diff --git a/resources/cider-ui-tests/views/main.ejs b/resources/cider-ui-tests/views/main.ejs
index b9b0ba11..852df620 100644
--- a/resources/cider-ui-tests/views/main.ejs
+++ b/resources/cider-ui-tests/views/main.ejs
@@ -75,8 +75,8 @@
                                             :max="mk.currentPlaybackDuration" :value="playerLCD.playbackDuration">
                                     </div>
                                 </div>
-                                <template v-if="mk.nowPlayingItem['attributes']['playParams']['id']">
-                                    <div class="actions" v-if="isInLibrary(mk.nowPlayingItem['attributes']['playParams']['id'])">❤️</div>
+                                <template v-if="mk.nowPlayingItem['attributes']['playParams']">
+                                    <div class="actions" v-if="isInLibrary(mk.nowPlayingItem['attributes']['playParams'])">❤️</div>
                                     <div class="actions" v-else>🖤</div>
                                 </template>