From 1add1a2233d73ae96f88deacf8db92b854f96f08 Mon Sep 17 00:00:00 2001
From: Thomas Fournier <thomas@artivain.com>
Date: Sun, 16 Jul 2023 23:55:35 -0400
Subject: [PATCH 1/2] change css query to target upgrade button

---
 preload.js | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/preload.js b/preload.js
index c91b0722..e5efed5c 100644
--- a/preload.js
+++ b/preload.js
@@ -139,7 +139,7 @@ function onApiLoaded() {
 
 	// Remove upgrade button
 	if (config.get("options.removeUpgradeButton")) {
-		const upgradeButton = $('ytmusic-pivot-bar-item-renderer[tab-id="SPunlimited"]')
+		const upgradeButton = $('ytmusic-guide-section-renderer #items:last-child') // Last child of #items
 		if (upgradeButton) {
 			upgradeButton.style.display = "none";
 		}

From 48b0469a4e5da68380e9123b78e2fbc8d2d988dc Mon Sep 17 00:00:00 2001
From: Thomas Fournier <thomas@artivain.com>
Date: Mon, 17 Jul 2023 00:42:33 -0400
Subject: [PATCH 2/2] Fix Remove upgrade button For some reason now it loads
 later, so I just added some CSS to hide it in the DOM instead of the element

---
 preload.js | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/preload.js b/preload.js
index e5efed5c..84f7272b 100644
--- a/preload.js
+++ b/preload.js
@@ -139,10 +139,13 @@ function onApiLoaded() {
 
 	// Remove upgrade button
 	if (config.get("options.removeUpgradeButton")) {
-		const upgradeButton = $('ytmusic-guide-section-renderer #items:last-child') // Last child of #items
-		if (upgradeButton) {
-			upgradeButton.style.display = "none";
-		}
+		const styles = document.createElement("style");
+		styles.innerHTML = `
+			ytmusic-guide-section-renderer #items ytmusic-guide-entry-renderer:last-child {
+				display: none;
+			}
+		`;
+		document.head.appendChild(styles);
 	}