diff --git a/packages/plugins/documentation/admin/src/components/Header/index.js b/packages/plugins/documentation/admin/src/components/Header/index.js
index 4993174a9c..96d62cc48e 100644
--- a/packages/plugins/documentation/admin/src/components/Header/index.js
+++ b/packages/plugins/documentation/admin/src/components/Header/index.js
@@ -1,7 +1,7 @@
 import React from 'react';
 import PropTypes from 'prop-types';
 import { useIntl } from 'react-intl';
-import { Header as Base } from '@buffetjs/custom';
+import { Header as BaseHeader } from '@buffetjs/custom';
 import { Button } from '@buffetjs/core';
 import { CheckPermissions } from '@strapi/helper-plugin';
 import openWithNewTab from '../../utils/openWithNewTab';
@@ -37,7 +37,6 @@ const Header = ({ currentDocVersion, docPrefixURL }) => {
         defaultMessage: 'Update',
       }),
       color: 'success',
-      // onClick: () => {},
       type: 'submit',
       key: 'button-submit',
       Component: props => (
@@ -49,7 +48,7 @@ const Header = ({ currentDocVersion, docPrefixURL }) => {
   ];
 
   return (
-    <Base
+    <BaseHeader
       actions={headerActions}
       content={formatMessage({
         id: getTrad('containers.HomePage.PluginHeader.description'),
diff --git a/packages/plugins/documentation/admin/src/components/Row/ButtonContainer.js b/packages/plugins/documentation/admin/src/components/Row/ButtonContainer.js
index 0623a6166b..a6eeab7cde 100755
--- a/packages/plugins/documentation/admin/src/components/Row/ButtonContainer.js
+++ b/packages/plugins/documentation/admin/src/components/Row/ButtonContainer.js
@@ -11,6 +11,16 @@ const ButtonContainer = ({ currentDocVersion, isHeader, onClick, onClickDelete,
     return <div />;
   }
 
+  const handleClickGenerate = e => {
+    e.preventDefault();
+    onClick(version);
+  };
+
+  const handleDelete = e => {
+    e.preventDefault();
+    onClickDelete(version);
+  };
+
   return (
     <div>
       <StyledButton
@@ -26,22 +36,13 @@ const ButtonContainer = ({ currentDocVersion, isHeader, onClick, onClickDelete,
         <StyledButton
           // FIXME when new DS ready
           type="generateDocumentation"
-          onClick={e => {
-            e.preventDefault();
-            onClick(version);
-          }}
+          onClick={handleClickGenerate}
         >
           <FormattedMessage id="documentation.components.Row.regenerate" />
         </StyledButton>
       </CheckPermissions>
       <CheckPermissions permissions={pluginPermissions.update}>
-        <StyledButton
-          type={version === currentDocVersion ? '' : 'trash'}
-          onClick={e => {
-            e.preventDefault();
-            onClickDelete(version);
-          }}
-        />
+        <StyledButton type={version === currentDocVersion ? '' : 'trash'} onClick={handleDelete} />
       </CheckPermissions>
     </div>
   );