diff --git a/src/DocSearchButton.tsx b/src/DocSearchButton.tsx
index ea033dff..aca6e887 100644
--- a/src/DocSearchButton.tsx
+++ b/src/DocSearchButton.tsx
@@ -32,7 +32,12 @@ export const DocSearchButton = React.forwardRef<
   }, []);
 
   return (
-    <button type="button" className="DocSearch-Button" {...props} ref={ref}>
+    <button
+      type="button"
+      className="DocSearch DocSearch-Button"
+      {...props}
+      ref={ref}
+    >
       <SearchIcon />
       <span className="DocSearch-Button-Placeholder">Search</span>
 
diff --git a/src/DocSearchModal.tsx b/src/DocSearchModal.tsx
index 992de797..7755595e 100644
--- a/src/DocSearchModal.tsx
+++ b/src/DocSearchModal.tsx
@@ -331,6 +331,7 @@ export function DocSearchModal({
         'aria-expanded': true,
       })}
       className={[
+        'DocSearch',
         'DocSearch-Container',
         state.status === 'stalled' && 'DocSearch-Container--Stalled',
         state.status === 'error' && 'DocSearch-Container--Errored',