diff --git a/app/controllers/budgets/investments_controller.rb b/app/controllers/budgets/investments_controller.rb index c17a356cb..25425250f 100644 --- a/app/controllers/budgets/investments_controller.rb +++ b/app/controllers/budgets/investments_controller.rb @@ -84,7 +84,8 @@ module Budgets end def load_ballot - @ballot = Budget::Ballot.where(user: current_user, budget: @budget).first_or_create + query = Budget::Ballot.where(user: current_user, budget: @budget) + @ballot = @budget.balloting? ? query.first_or_create : query.first_or_initialize end def load_heading