From 2c5db670a4b64a46a50ed065b06bde67b46174cc Mon Sep 17 00:00:00 2001
From: Ajay <dev@ajay.app>
Date: Fri, 4 Nov 2022 15:44:14 -0400
Subject: [PATCH] Safety on getChapterButton

Fix #1573
---
 src/js-components/previewBar.ts | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/js-components/previewBar.ts b/src/js-components/previewBar.ts
index c9ece9e7..3eda3b3d 100644
--- a/src/js-components/previewBar.ts
+++ b/src/js-components/previewBar.ts
@@ -740,7 +740,7 @@ class PreviewBar {
                 || ((!segments || segments.length <= 0) && submittingSegments?.length <= 0)) {
             const chaptersContainer = this.getChaptersContainer();
             const chapterButton = this.getChapterButton(chaptersContainer);
-            if (chapterButton.classList.contains("ytp-chapter-container-disabled")) {
+            if (chapterButton && chapterButton.classList.contains("ytp-chapter-container-disabled")) {
                 chaptersContainer.style.display = "none";
             }
 
@@ -820,7 +820,7 @@ class PreviewBar {
 
     private getChapterButton(chaptersContainer: HTMLElement): HTMLButtonElement {
         return (chaptersContainer ?? this.getChaptersContainer())
-            .querySelector("button.ytp-chapter-title") as HTMLButtonElement;
+            ?.querySelector("button.ytp-chapter-title") as HTMLButtonElement;
     }
 
     remove(): void {