diff --git a/docs-site/src/examples/customInput.js b/docs-site/src/examples/customInput.js
index 0838e982f..731200311 100644
--- a/docs-site/src/examples/customInput.js
+++ b/docs-site/src/examples/customInput.js
@@ -1,15 +1,17 @@
 () => {
   const [startDate, setStartDate] = useState(new Date());
-  const ExampleCustomInput = forwardRef(({ value, onClick }, ref) => (
-    <button className="example-custom-input" onClick={onClick} ref={ref}>
-      {value}
-    </button>
-  ));
+  const ExampleCustomInput = forwardRef(
+    ({ value, onClick, className }, ref) => (
+      <button className={className} onClick={onClick} ref={ref}>
+        {value}
+      </button>
+    ),
+  );
   return (
     <DatePicker
       selected={startDate}
       onChange={(date) => setStartDate(date)}
-      customInput={<ExampleCustomInput />}
+      customInput={<ExampleCustomInput className="example-custom-input" />}
     />
   );
 };