diff --git a/src/button/button-contributions.ts b/src/button/button-contributions.ts
index f8c7f35..3f87256 100644
--- a/src/button/button-contributions.ts
+++ b/src/button/button-contributions.ts
@@ -191,7 +191,7 @@ export const buttonContributions: ButtonContributionParams[] = [
         ],
         // must not match /blob/ because that is a file
         match: /^(?!.*\/blob\/).*$/,
-        selector: "#tree-holder > div > div.tree-controls > div:first-child",
+        selector: "#tree-holder > section > div > div.tree-controls",
         containerElement: { type: "div", props: { marginLeft: "8px" } },
         application: "gitlab",
         manipulations: [
diff --git a/test/src/button-contributions-copy.ts b/test/src/button-contributions-copy.ts
index f8c7f35..3f87256 100644
--- a/test/src/button-contributions-copy.ts
+++ b/test/src/button-contributions-copy.ts
@@ -191,7 +191,7 @@ export const buttonContributions: ButtonContributionParams[] = [
         ],
         // must not match /blob/ because that is a file
         match: /^(?!.*\/blob\/).*$/,
-        selector: "#tree-holder > div > div.tree-controls > div:first-child",
+        selector: "#tree-holder > section > div > div.tree-controls",
         containerElement: { type: "div", props: { marginLeft: "8px" } },
         application: "gitlab",
         manipulations: [