diff --git a/docs/source/_static/scripts/custom.js b/docs/source/_static/scripts/custom.js
index 0f718376..eb8400dc 100644
--- a/docs/source/_static/scripts/custom.js
+++ b/docs/source/_static/scripts/custom.js
@@ -188,20 +188,10 @@ const renderActionsList = () => {
   })
 }
 
-const addFilterBar = () => {
-  const container = document.querySelector('.bd-main')
-  if (!container)
-    return
-
-  const referenceSection = document.getElementById('reference')
-  if (!referenceSection)
-    return
-
-  const header = referenceSection.querySelector('h2')
-  if (!header)
-    return
-
+const createFilterBar = () => {
   const input = document.createElement('input')
+  const referenceSection = document.getElementById('reference')
+
   input.type = 'text'
   input.placeholder = 'Filter'
   input.classList.add('filter-bar')
@@ -216,13 +206,38 @@ const addFilterBar = () => {
     })
   })
 
-  // Apply the fixed class if the header is above the viewport
-  const headerOffsetTop = header.offsetTop
-  document.addEventListener('scroll', () => {
-    header.classList.toggle('fixed', headerOffsetTop < window.scrollY)
-  })
+  return input
+}
 
+const addFilterBar = () => {
+  const container = document.querySelector('.bd-main')
+  if (!container)
+    return
+
+  const referenceSection = document.getElementById('reference')
+  if (!referenceSection)
+    return
+
+  const header = referenceSection.querySelector('h2')
+  if (!header)
+    return
+
+  const origInnerHTML = header.innerHTML
+  header.innerHTML = '<span class="header-content">' + origInnerHTML + '</span>'
+
+  const input = createFilterBar()
   header.appendChild(input)
+
+  const headerOffsetTop = header.offsetTop
+
+  // Apply the fixed class if the header is above the viewport
+  document.addEventListener('scroll', () => {
+    if (headerOffsetTop < window.scrollY) {
+      header.classList.add('fixed')
+    } else {
+      header.classList.remove('fixed')
+    }
+  })
 }
 
 document.addEventListener("DOMContentLoaded", function() {
diff --git a/docs/source/_static/styles/custom.css b/docs/source/_static/styles/custom.css
index 797de928..03056de3 100644
--- a/docs/source/_static/styles/custom.css
+++ b/docs/source/_static/styles/custom.css
@@ -156,7 +156,10 @@ ul.grid .icon {
   top: 0;
   background: white;
   z-index: 1;
-  border-bottom: 1px solid #ccc;
+}
+
+#reference h2.fixed .header-content {
+  display: none;
 }
 
 @media screen and (max-width: 768px) {