diff --git a/extension/script.js b/extension/script.js
index 42d1e61..e2fc3d6 100644
--- a/extension/script.js
+++ b/extension/script.js
@@ -437,11 +437,13 @@ function checkVideoExists(taButton) {
     console.error(e);
   }
 
-  if (!taButton.parentElement) return;
-  let videoId = getVideoId(taButton.parentElement);
-  taButton.setAttribute('data-id', videoId);
-  taButton.setAttribute('data-type', 'video');
-  taButton.title = `TA download video: ${taButton.parentElement.innerText} [${videoId}]`;
+  let videoId = taButton.dataset.id;
+  if (taButton.parentElement) {
+    videoId = getVideoId(taButton.parentElement);
+    taButton.setAttribute('data-id', videoId);
+    taButton.setAttribute('data-type', 'video');
+    taButton.title = `TA download video: ${taButton.parentElement.innerText} [${videoId}]`;
+  }
 
   let message = { type: 'videoExists', videoId };
   let sending = sendMessage(message);