From 186938216639493c5dbede4d437d146761f5c946 Mon Sep 17 00:00:00 2001
From: Ajay Ramachandran <dev@ajay.app>
Date: Tue, 18 Feb 2020 19:03:34 -0500
Subject: [PATCH] Fixed popup issues.

---
 src/content.ts |  3 +++
 src/popup.ts   | 23 ++++++++++++-----------
 2 files changed, 15 insertions(+), 11 deletions(-)

diff --git a/src/content.ts b/src/content.ts
index 2c5a413e..d3ecfa99 100644
--- a/src/content.ts
+++ b/src/content.ts
@@ -953,6 +953,9 @@ function updateSponsorTimesSubmitting() {
                 sponsorTimesSubmitting = sponsorTimes;
 
                 updatePreviewBar();
+
+                // Restart skipping schedule
+                startSponsorSchedule();
             }
         }
     });
diff --git a/src/popup.ts b/src/popup.ts
index c61a0a5d..c29f486a 100644
--- a/src/popup.ts
+++ b/src/popup.ts
@@ -719,17 +719,8 @@ async function runThePopup(messageListener?: MessageListener) {
         sponsorTimes.splice(index, 1);
   
         //save this
-		Config.config.sponsorTimes.set(currentVideoID, sponsorTimes);
-            messageHandler.query({
-                active: true,
-                currentWindow: true
-            }, tabs => {
-                messageHandler.sendMessage(
-                    tabs[0].id,
-                    {message: "sponsorDataChanged"}
-                );
-            });
-  
+        Config.config.sponsorTimes.set(currentVideoID, sponsorTimes);
+        
         //update display
         displaySponsorTimes();
   
@@ -750,6 +741,16 @@ async function runThePopup(messageListener?: MessageListener) {
             //hide submission section
             document.getElementById("submissionSection").style.display = "none";
         }
+
+        messageHandler.query({
+            active: true,
+            currentWindow: true
+        }, tabs => {
+            messageHandler.sendMessage(
+                tabs[0].id,
+                {message: "sponsorDataChanged"}
+            );
+        });
     }
   
     function clearTimes() {