Extract method to get investment milestone status

app/controllers/budgets/executions_controller.rb

@@ -15,9 +15,7 @@ module Budgets
         if params[:status].present?
           @budget.investments.winners
                   .joins(:milestones).includes(:milestones)
-                  .select { |i| i.milestones.published.with_status
-                                            .order_by_publication_date.last
-                                            .try(:status_id) == params[:status].to_i }
+                  .select { |i| i.milestone_status_id == params[:status].to_i }
                   .uniq
                   .group_by(&:heading)
         else

app/helpers/budget_executions_helper.rb

@@ -1,9 +1,9 @@
 module BudgetExecutionsHelper
 
   def filters_select_counts(status)
-    @budget.investments.winners.with_milestones.select { |i| i.milestones
-                                      .published.with_status.order_by_publication_date
-                                      .last.status_id == status rescue false }.count
+    @budget.investments.winners.with_milestones.select do |investment|
+      investment.milestone_status_id == status
+    end.count
   end
 
   def first_milestone_with_image(investment)

app/models/budget/investment.rb

@@ -342,6 +342,10 @@ class Budget
       self.valuator_groups.collect(&:name).compact.join(', ').presence
     end
 
+    def milestone_status_id
+      milestones.published.with_status.order_by_publication_date.last&.status_id
+    end
+
     private
 
       def set_denormalized_ids