From 059a674ae1f7db9868e3d4af59c3770a54a2e721 Mon Sep 17 00:00:00 2001
From: Michael C <michael@mchang.name>
Date: Sat, 15 Jan 2022 22:49:22 -0500
Subject: [PATCH] fix hover preview segments

---
 src/content.ts | 2 +-
 src/utils.ts   | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/content.ts b/src/content.ts
index 81ba097b..0936d238 100644
--- a/src/content.ts
+++ b/src/content.ts
@@ -945,7 +945,7 @@ function getYouTubeVideoID(document: Document): string | boolean {
     // skip to document if matches pattern
     if (url.includes("/channel/") || url.includes("/user/") || url.includes("/c/")) return getYouTubeVideoIDFromDocument(document);
     // not sure, try URL then document
-    return getYouTubeVideoIDFromURL(url) || getYouTubeVideoIDFromDocument(document);
+    return getYouTubeVideoIDFromURL(url) || getYouTubeVideoIDFromDocument(document, false);
 }
 
 function getYouTubeVideoIDFromDocument(document: Document, hideIcon = true): string | boolean {
diff --git a/src/utils.ts b/src/utils.ts
index cbbf03a3..a2687f3d 100644
--- a/src/utils.ts
+++ b/src/utils.ts
@@ -331,7 +331,7 @@ export default class Utils {
 
     findReferenceNode(): HTMLElement {
         const selectors = [
-            "#player-container-id",
+            "#player-container",
             "#movie_player",
             "#c4-player", // Channel Trailer
             "#main-panel.ytmusic-player-page", // YouTube music
@@ -347,7 +347,7 @@ export default class Utils {
                 let index = 1;
 
                 //find the child that is the video player (sometimes it is not the first)
-                while (index < player.children.length && (!referenceNode.classList.contains("html5-video-player") || !referenceNode.classList.contains("ytp-embed"))) {
+                while (index < player.children.length && (!referenceNode.classList?.contains("html5-video-player") || !referenceNode.classList?.contains("ytp-embed"))) {
                     referenceNode = player.children[index] as HTMLElement;
 
                     index++;