diff --git a/assets/css/js_interop.css b/assets/css/js_interop.css
index d7a10d52e..bcd45b68c 100644
--- a/assets/css/js_interop.css
+++ b/assets/css/js_interop.css
@@ -112,6 +112,18 @@ solely client-side operations.
   left: calc(-45vw + 50%);
 }
 
+[data-element="cell"][data-js-amplified]
+  [data-element="amplify-outputs-button"]
+  [data-element="zoom-in-icon"] {
+  @apply hidden;
+}
+
+[data-element="cell"]:not([data-js-amplified])
+  [data-element="amplify-outputs-button"]
+  [data-element="zoom-out-icon"] {
+  @apply hidden;
+}
+
 [data-element="session"]:not([data-js-side-panel-content])
   [data-element="side-panel"] {
   @apply hidden;
diff --git a/lib/livebook_web/live/session_live/cell_component.ex b/lib/livebook_web/live/session_live/cell_component.ex
index d2c12e071..7954a5d52 100644
--- a/lib/livebook_web/live/session_live/cell_component.ex
+++ b/lib/livebook_web/live/session_live/cell_component.ex
@@ -99,7 +99,8 @@ defmodule LivebookWeb.SessionLive.CellComponent do
         <span class="tooltip top" data-tooltip="Amplify output" data-element="amplify-outputs-button">
           <button class="icon-button"
             aria-label="amplify outputs">
-            <.remix_icon icon="zoom-in-line" class="text-xl" />
+            <.remix_icon icon="zoom-in-line" class="text-xl" data-element="zoom-in-icon" />
+            <.remix_icon icon="zoom-out-line" class="text-xl" data-element="zoom-out-icon" />
           </button>
         </span>
         <.cell_settings_button cell_id={@cell_view.id} socket={@socket} session_id={@session_id} />