diff --git a/app/controllers/valuation/budget_investments_controller.rb b/app/controllers/valuation/budget_investments_controller.rb
index 9a05f3cc8..e9cd4f21e 100644
--- a/app/controllers/valuation/budget_investments_controller.rb
+++ b/app/controllers/valuation/budget_investments_controller.rb
@@ -87,7 +87,7 @@ class Valuation::BudgetInvestmentsController < Valuation::BaseController
         filters << {
                      name: heading.name,
                      id: heading.id,
-                     count: investments.select { |i| i.heading_id == heading.id }.size
+                     count: investments.count { |i| i.heading_id == heading.id }
                    }
       end
     end