setting segment is before node rendering

hanydd · Feb 4, 2025 · e431d8b · e431d8b
1 parent 1c25ccd
commit e431d8b
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/src/render/CategoryPill.tsx b/src/render/CategoryPill.tsx
@@ -32,8 +32,7 @@ export class CategoryPill {
         // bilibili will set the textContent of the title after loading for some reason.
         // If the node is inserted before this reset of title, it will be removed
         const mutationCounter = () => (this.mutationCount += 1);
-        mutationCounter.bind(this);
-        this.mutationObserver = new MutationObserver(mutationCounter);
+        this.mutationObserver = new MutationObserver(mutationCounter.bind(this));
 
         addCleanupListener(() => {
             if (this.mutationObserver) {
@@ -117,8 +116,6 @@ export class CategoryPill {
     }
 
     async setSegment(segment: SponsorTime): Promise<void> {
-        await waitFor(() => this.ref.current);
-
         if (this.ref.current?.state?.segment !== segment) {
             const newState = {
                 segment,