diff --git a/app/views/budgets/investments/_ballot.html.erb b/app/views/budgets/investments/_ballot.html.erb
index 32a788756..bbac474af 100644
--- a/app/views/budgets/investments/_ballot.html.erb
+++ b/app/views/budgets/investments/_ballot.html.erb
@@ -1,6 +1,6 @@
-<% reason = investment.reason_for_not_being_ballotable_by(current_user, investment_ballot) %>
+<% reason = investment.reason_for_not_being_ballotable_by(current_user, ballot) %>
- <% if investment_ballot.has_investment?(investment) %>
+ <% if ballot.has_investment?(investment) %>
- <% if reason.present? && !investment_ballot.has_investment?(investment) %>
+ <% if reason.present? && !ballot.has_investment?(investment) %>
diff --git a/app/views/budgets/investments/_investment.html.erb b/app/views/budgets/investments/_investment.html.erb
index 330c3b690..87c2f663e 100644
--- a/app/views/budgets/investments/_investment.html.erb
+++ b/app/views/budgets/investments/_investment.html.erb
@@ -62,7 +62,7 @@
<%= render partial: '/budgets/investments/ballot', locals: {
investment: investment,
investment_ids: investment_ids,
- investment_ballot: investment_ballots[investment.budget]
+ ballot: ballot
} %>
diff --git a/app/views/budgets/investments/_investment_show.html.erb b/app/views/budgets/investments/_investment_show.html.erb
index d5d371290..96fc74681 100644
--- a/app/views/budgets/investments/_investment_show.html.erb
+++ b/app/views/budgets/investments/_investment_show.html.erb
@@ -65,7 +65,11 @@
<% elsif investment.should_show_ballots? %>
- <%= render 'ballot', investment: investment %>
+ <%= render partial: 'ballot', locals: {
+ investment: investment,
+ investment_ids: investment_ids,
+ ballot: ballot,
+ } %>
<% end %>
diff --git a/app/views/budgets/investments/index.html.erb b/app/views/budgets/investments/index.html.erb
index 622921eab..967bbb2e3 100644
--- a/app/views/budgets/investments/index.html.erb
+++ b/app/views/budgets/investments/index.html.erb
@@ -40,7 +40,7 @@
investment: investment,
investment_ids: @investment_ids,
investment_votes: @investment_votes,
- investment_ballots: @investment_ballots
+ ballot: @ballot
} %>
<% end %>
diff --git a/app/views/budgets/investments/show.html.erb b/app/views/budgets/investments/show.html.erb
index 6d4b735e1..be33605b0 100644
--- a/app/views/budgets/investments/show.html.erb
+++ b/app/views/budgets/investments/show.html.erb
@@ -2,7 +2,9 @@
<%= render partial: '/budgets/investments/investment_show', locals: {
investment: @investment,
- investment_votes: @investment_votes
+ investment_ids: @investment_ids,
+ investment_votes: @investment_votes,
+ ballot: @ballot
} %>
<%= render partial: '/comments/comment_tree', locals: { comment_tree: @comment_tree, comment_flags: @comment_flags } %>
diff --git a/app/views/management/budgets/investments/index.html.erb b/app/views/management/budgets/investments/index.html.erb
index ad48c0578..41edb7bcd 100644
--- a/app/views/management/budgets/investments/index.html.erb
+++ b/app/views/management/budgets/investments/index.html.erb
@@ -22,7 +22,7 @@
investment: investment,
investment_ids: @investment_ids,
investment_votes: @investment_votes,
- investment_ballots: @investment_ballots
+ ballot: @ballot
} %>
<% end %>
diff --git a/app/views/management/budgets/investments/print.html.erb b/app/views/management/budgets/investments/print.html.erb
index d2390879b..6bdd84713 100644
--- a/app/views/management/budgets/investments/print.html.erb
+++ b/app/views/management/budgets/investments/print.html.erb
@@ -27,7 +27,7 @@
investment: investment,
investment_ids: @investment_ids,
investment_votes: @investment_votes,
- investment_ballots: @investment_ballots
+ ballot: @ballot
} %>
<% end %>