diff --git a/app/components/admin/budget_headings/headings_component.html.erb b/app/components/admin/budget_headings/headings_component.html.erb
index 189d7166d..cde24117b 100644
--- a/app/components/admin/budget_headings/headings_component.html.erb
+++ b/app/components/admin/budget_headings/headings_component.html.erb
@@ -28,6 +28,6 @@
<% else %>
- <%= t("admin.budget_headings.no_headings") %>
+ <%= sanitize(t("admin.budget_headings.no_headings", group: group.name)) %>
<% end %>
diff --git a/app/components/admin/budget_headings/headings_component.rb b/app/components/admin/budget_headings/headings_component.rb
index 3b6594f23..396944392 100644
--- a/app/components/admin/budget_headings/headings_component.rb
+++ b/app/components/admin/budget_headings/headings_component.rb
@@ -8,10 +8,10 @@ class Admin::BudgetHeadings::HeadingsComponent < ApplicationComponent
private
def group
- @group ||= headings.first.group
+ @group ||= headings.proxy_association.owner
end
def budget
- @budget ||= headings.first.budget
+ @budget ||= group.budget
end
end
diff --git a/app/components/admin/budgets_wizard/headings/group_switcher_component.html.erb b/app/components/admin/budgets_wizard/headings/group_switcher_component.html.erb
index 4fe2ba830..1c04c5af5 100644
--- a/app/components/admin/budgets_wizard/headings/group_switcher_component.html.erb
+++ b/app/components/admin/budgets_wizard/headings/group_switcher_component.html.erb
@@ -1,12 +1,12 @@
<% if other_groups.one? %>
- <%= t("admin.budget_headings.group_switcher.currently_showing", group: group.name) %>
+ <%= currently_showing_text %>
<%= link_to t("admin.budget_headings.group_switcher.the_other_group", group: other_groups.first.name),
headings_path(other_groups.first) %>
<% else %>
-
<%= t("admin.budget_headings.group_switcher.currently_showing", group: group.name) %>
+
<%= currently_showing_text %>