diff --git a/app/components/lookbook/display_options/editor/component.html.erb b/app/components/lookbook/display_options/editor/component.html.erb
index a09a89f97..fb795c3fd 100644
--- a/app/components/lookbook/display_options/editor/component.html.erb
+++ b/app/components/lookbook/display_options/editor/component.html.erb
@@ -1,7 +1,7 @@
<%= render_component_tag class: "px-2" do %>
<% if fields.many? %>
<%= lookbook_render :button, icon: :settings, tooltip: "Display options" do |button| %>
- <% button.dropdown do %>
+ <% button.with_dropdown do %>
<%= safe_join(fields) %>
diff --git a/app/components/lookbook/embed/component.html.erb b/app/components/lookbook/embed/component.html.erb
index 54b49b4ef..8d8628c20 100644
--- a/app/components/lookbook/embed/component.html.erb
+++ b/app/components/lookbook/embed/component.html.erb
@@ -2,17 +2,17 @@
<%= render_component_tag class: "not-prose border-b border-lookbook-divider rounded-sm overflow-hidden", "@navigation:start.window": "cleanup" do %>
<%= lookbook_render :toolbar, class: "border border-b-0 border-lookbook-divider" do |toolbar| %>
- <% toolbar.section padded: true do %>
+ <% toolbar.with_section padded: true do %>
<%= @target.preview.label %> (<%= @target.label %>)
<% end %>
- <% toolbar.section align: :right, divide: :left do %>
+ <% toolbar.with_section align: :right, divide: :left do %>
<%= lookbook_render :button_group do |group| %>
- <% group.button icon: :eye,
+ <% group.with_button icon: :eye,
href: lookbook_inspect_path(@target.path, @params),
tooltip: "View in Inspector" %>
- <% group.button icon: :external_link,
+ <% group.with_button icon: :external_link,
href: lookbook_preview_path(@target.path, @params),
tooltip: "Open in new window",
target: "_blank" %>
diff --git a/app/components/lookbook/header/component.html.erb b/app/components/lookbook/header/component.html.erb
index 9a226ddea..0eeb9e0da 100644
--- a/app/components/lookbook/header/component.html.erb
+++ b/app/components/lookbook/header/component.html.erb
@@ -1,6 +1,6 @@
<%= render_component_tag :header do %>
<%= lookbook_render :toolbar, class: "!bg-lookbook-header-bg !text-lookbook-header-text !border-lookbook-header-border" do |toolbar| %>
- <% toolbar.section padded: true do %>
+ <% toolbar.with_section padded: true do %>
<% if branding.present? %>
href="<%= landing_path %>"<% end %>
@@ -10,7 +10,7 @@
<% end %>
<% end %>
- <% toolbar.section padded: false, align: :right, class: "flex items-center" do %>
+ <% toolbar.with_section padded: false, align: :right, class: "flex items-center" do %>