diff --git a/lib/livebook_web/live/session_live/custom_view_component.ex b/lib/livebook_web/live/session_live/custom_view_component.ex
index 5c45a6441..24b23770e 100644
--- a/lib/livebook_web/live/session_live/custom_view_component.ex
+++ b/lib/livebook_web/live/session_live/custom_view_component.ex
@@ -4,7 +4,7 @@ defmodule LivebookWeb.SessionLive.CustomViewComponent do
   @impl true
   def render(assigns) do
     ~H"""
-    <div class="p-6">
+    <div class="flex flex-col">
       <h3 class="text-2xl font-semibold text-gray-800">
         Custom view
       </h3>