diff --git a/src/components/pipeline-warning/pipeline-warning.js b/src/components/pipeline-warning/pipeline-warning.js
index ccf8b6aca2..9071f6ff86 100644
--- a/src/components/pipeline-warning/pipeline-warning.js
+++ b/src/components/pipeline-warning/pipeline-warning.js
@@ -47,10 +47,12 @@ export const PipelineWarning = ({
             a while to render. You can use the sidebar controls to select a
             smaller graph.
           </p>
-          <Button onClick={onHide}>Render it anyway</Button>
-          <Button mode="secondary" onClick={onDisable} size="small">
-            Don't show this again
-          </Button>
+          <div className="pipeline-warning__button-wrapper">
+            <Button onClick={onHide}>Render it anyway</Button>
+            <Button mode="secondary" onClick={onDisable} size="small">
+              Don't show this again
+            </Button>
+          </div>
         </div>
       )}
       {isEmptyPipeline && componentLoaded && (
@@ -70,14 +72,9 @@ export const PipelineWarning = ({
       )}
       {invalidUrl && componentLoaded && (
         <div
-          className={classnames(
-            'kedro',
-            'pipeline-warning',
-            'pipeline-warning--invalid-url',
-            {
-              'pipeline-warning--sidebar-visible': sidebarVisible,
-            }
-          )}
+          className={classnames('kedro', 'pipeline-warning', {
+            'pipeline-warning--sidebar-visible': sidebarVisible,
+          })}
         >
           <h2 className="pipeline-warning__title">
             Oops, this URL isn't valid
diff --git a/src/components/pipeline-warning/pipeline-warning.scss b/src/components/pipeline-warning/pipeline-warning.scss
index d964548b07..37b4e182c7 100644
--- a/src/components/pipeline-warning/pipeline-warning.scss
+++ b/src/components/pipeline-warning/pipeline-warning.scss
@@ -74,8 +74,3 @@
 .pipeline-warning .button:first-of-type {
   margin-bottom: 1.2em;
 }
-
-.pipeline-warning--invalid-url .button {
-  display: flex;
-  justify-content: center;
-}
diff --git a/src/components/ui/button/button.scss b/src/components/ui/button/button.scss
index 7c992e26ff..1e6355b9e8 100644
--- a/src/components/ui/button/button.scss
+++ b/src/components/ui/button/button.scss
@@ -35,7 +35,6 @@ $secondary-underline-offset-hover: 4px;
   box-shadow: none;
   color: var(--color-button__text);
   cursor: pointer;
-  display: flex;
   font-family: inherit;
   font-size: 1.6em;
   font-weight: 600;
@@ -139,7 +138,6 @@ $secondary-underline-offset-hover: 4px;
 
 .success-check-mark {
   animation: grow 0.35s 0.4s ease forwards;
-  display: block;
   margin-left: 0.5em;
   transform: scale(0);
 }