From ef0acaf20f609bd50767cb83f6481314f641ef4f Mon Sep 17 00:00:00 2001 From: kikito Date: Sun, 25 Dec 2016 12:54:32 +0100 Subject: [PATCH] Only creates ballot when budget balloting --- app/controllers/budgets/investments_controller.rb | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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