diff --git a/source/features/clean-conversation-headers.tsx b/source/features/clean-conversation-headers.tsx index e5154fd6af9c..a7e1342a3da7 100644 --- a/source/features/clean-conversation-headers.tsx +++ b/source/features/clean-conversation-headers.tsx @@ -4,7 +4,7 @@ import React from 'dom-chef'; import elementReady from 'element-ready'; import * as pageDetect from 'github-url-detection'; import ArrowLeftIcon from 'octicons-plain-react/ArrowLeft'; -import {$, closestElementOptional} from 'select-dom'; +import {$, closestElement, closestElementOptional} from 'select-dom'; import features from '../feature-manager.js'; import getDefaultBranch from '../github-helpers/get-default-branch.js'; @@ -90,7 +90,8 @@ function replaceFromWithArrow(base: HTMLElement): void { ); } -async function cleanPrHeader(summaryRow: HTMLElement): Promise { +async function cleanPrHeader(headRef: HTMLElement): Promise { + const summaryRow = closestElement('.d-flex[class*="PullRequestHeaderSummary"]', headRef); summaryRow.classList.add('rgh-clean-conversation-headers'); if (isStickyHeader(summaryRow)) { @@ -122,7 +123,7 @@ async function cleanPrHeader(summaryRow: HTMLElement): Promise { async function init(signal: AbortSignal): Promise { observe( - '.d-flex[class*="PullRequestHeaderSummary"]', + '[class^="PullRequestBranchName"] ~ div [class^="PullRequestBranchName"]', cleanPrHeader, {signal}, );