diff --git a/src/features/dsp408/components/GainSection/GainSection.tsx b/src/features/dsp408/components/GainSection/GainSection.tsx
index 30683d6..1181a9a 100644
--- a/src/features/dsp408/components/GainSection/GainSection.tsx
+++ b/src/features/dsp408/components/GainSection/GainSection.tsx
@@ -431,8 +431,8 @@ function GainSection({
             <div className="gain-graph-title-line">
               <div className="gain-graph-title">Input Response</div>
               <Button
-                variant={showInputStages ? 'outline' : 'ghost'}
-                className="gain-stages-toggle"
+                variant="toggle"
+                className={`gain-stages-toggle ${showInputStages ? 'active-info' : ''}`}
                 onClick={() => setShowInputStages((v) => !v)}
               >
                 Show stages
diff --git a/src/features/dsp408/components/GateSection/GateSection.css b/src/features/dsp408/components/GateSection/GateSection.css
index 709d447..9ac76e1 100644
--- a/src/features/dsp408/components/GateSection/GateSection.css
+++ b/src/features/dsp408/components/GateSection/GateSection.css
@@ -23,10 +23,12 @@
 }
 
 .gate-panels {
+  display: grid;
+  grid-auto-flow: column;
   height: 100%;
-  display: flex;
   gap: var(--space-2);
   overflow-x: auto;
+  justify-content: center;
 }
 
 .gate-panel {