diff --git a/app/components/budgets/ballot/ballot_component.html.erb b/app/components/budgets/ballot/ballot_component.html.erb
index 6834a3059..5945b7f2b 100644
--- a/app/components/budgets/ballot/ballot_component.html.erb
+++ b/app/components/budgets/ballot/ballot_component.html.erb
@@ -36,9 +36,7 @@
         <% end %>
 
         <ul class="ballot-list">
-          <%= render Budgets::Ballot::InvestmentComponent.with_collection(
-            ballot.investments.by_group(group.id)
-          ) %>
+          <%= render Budgets::Ballot::InvestmentComponent.with_collection(group_investments(group)) %>
         </ul>
       </div>
     </div>
diff --git a/app/components/budgets/ballot/ballot_component.rb b/app/components/budgets/ballot/ballot_component.rb
index fbb8b1d75..5b72f0082 100644
--- a/app/components/budgets/ballot/ballot_component.rb
+++ b/app/components/budgets/ballot/ballot_component.rb
@@ -26,4 +26,8 @@ class Budgets::Ballot::BallotComponent < ApplicationComponent
         budget_investments_path(budget, heading_id: group.headings.first)
       end
     end
+
+    def group_investments(group)
+      ballot.investments.by_group(group.id)
+    end
 end
diff --git a/app/components/budgets/investments/my_ballot_component.html.erb b/app/components/budgets/investments/my_ballot_component.html.erb
index 307be97cc..99ada7a05 100644
--- a/app/components/budgets/investments/my_ballot_component.html.erb
+++ b/app/components/budgets/investments/my_ballot_component.html.erb
@@ -3,7 +3,7 @@
     <%= t("budgets.investments.index.sidebar.my_ballot") %>
   </h2>
 
-  <% if ballot.investments.by_heading(heading.id).count > 0 %>
+  <% if investments.count > 0 %>
     <p>
       <em><%= sanitize(ballot.voted_info(heading)) %></em>
     </p>
@@ -32,7 +32,7 @@
   <ul class="ballot-list">
     <% if heading %>
       <%= render Budgets::Ballot::InvestmentForSidebarComponent.with_collection(
-        ballot.investments.by_heading(heading.id),
+        investments,
         investment_ids: investment_ids
       ) %>
     <% end %>
diff --git a/app/components/budgets/investments/my_ballot_component.rb b/app/components/budgets/investments/my_ballot_component.rb
index 0f7ea5ff5..05f23c5c6 100644
--- a/app/components/budgets/investments/my_ballot_component.rb
+++ b/app/components/budgets/investments/my_ballot_component.rb
@@ -18,4 +18,8 @@ class Budgets::Investments::MyBallotComponent < ApplicationComponent
     def budget
       ballot.budget
     end
+
+    def investments
+      ballot.investments.by_heading(heading.id)
+    end
 end