1 files changed, 17 insertions, 2 deletions
diff --git a/Extensions/UserScript/Return Youtube Dislike.user.js b/Extensions/UserScript/Return Youtube Dislike.user.js
index 253344c..c45efd1 100644
--- a/Extensions/UserScript/Return Youtube Dislike.user.js
+++ b/Extensions/UserScript/Return Youtube Dislike.user.js
@@ -94,10 +94,25 @@ function getLikeButton() {
   return getButtons().children[0];
 }
 
+function getLikeTextContainer() {
+  return (
+    getLikeButton().querySelector("#text") ??
+    getLikeButton().getElementsByTagName("yt-formatted-string")[0]
+  );
+}
+
 function getDislikeButton() {
   return getButtons().children[1];
 }
 
+function getDislikeTextContainer() {
+  return (
+    getDislikeButton().querySelector("#text") ??
+    getDislikeButton().getElementsByTagName("yt-formatted-string")[0]
+  );
+}
+
+
 let mutationObserver = new Object();
 
 if (isShorts() && mutationObserver.exists !== true) {
@@ -197,7 +212,7 @@ function setLikes(likesCount) {
       likesCount;
     return;
   }
-  getButtons().children[0].querySelector("#text").innerText = likesCount;
+  getLikeTextContainer().innerText = likesCount;
 }
 
 function setDislikes(dislikesCount) {
@@ -205,7 +220,7 @@ function setDislikes(dislikesCount) {
     mobileDislikes = dislikesCount;
     return;
   }
-  getButtons().children[1].querySelector("#text").innerText = dislikesCount;
+  getDislikeTextContainer().innerText = dislikesCount;
 }
 
 function getLikeCountFromButton() {