diff --git a/docs/data/tree-view/tree-item-customization/useTreeItemHookPublicAPI.js b/docs/data/tree-view/tree-item-customization/useTreeItemHookPublicAPI.js
index 33e6b70c748e..0c71983ff031 100644
--- a/docs/data/tree-view/tree-item-customization/useTreeItemHookPublicAPI.js
+++ b/docs/data/tree-view/tree-item-customization/useTreeItemHookPublicAPI.js
@@ -19,7 +19,12 @@ function CustomLabel({ children, className, selectFirstChildren }) {
     >
       <Typography>{children}</Typography>
       {!!selectFirstChildren && (
-        <Button size="small" onClick={selectFirstChildren}>
+        <Button
+          size="small"
+          variant="text"
+          sx={{ position: 'absolute', right: 0, top: 0 }}
+          onClick={selectFirstChildren}
+        >
           Select child
         </Button>
       )}
diff --git a/docs/data/tree-view/tree-item-customization/useTreeItemHookPublicAPI.tsx b/docs/data/tree-view/tree-item-customization/useTreeItemHookPublicAPI.tsx
index 0fad6d26966f..3698ce968f02 100644
--- a/docs/data/tree-view/tree-item-customization/useTreeItemHookPublicAPI.tsx
+++ b/docs/data/tree-view/tree-item-customization/useTreeItemHookPublicAPI.tsx
@@ -29,7 +29,12 @@ function CustomLabel({
     >
       <Typography>{children}</Typography>
       {!!selectFirstChildren && (
-        <Button size="small" onClick={selectFirstChildren}>
+        <Button
+          size="small"
+          variant="text"
+          sx={{ position: 'absolute', right: 0, top: 0 }}
+          onClick={selectFirstChildren}
+        >
           Select child
         </Button>
       )}