diff --git a/app/components/milestones/progress_bars_component.html.erb b/app/components/milestones/progress_bars_component.html.erb
index e07e71b8c..e069d5c6e 100644
--- a/app/components/milestones/progress_bars_component.html.erb
+++ b/app/components/milestones/progress_bars_component.html.erb
@@ -2,12 +2,12 @@
<%= t("milestones.index.progress") %>
- <%= progress_tag_for(milestoneable.primary_progress_bar) %>
+ <%= progress_tag_for(primary_progress_bar) %>
- <% if milestoneable.secondary_progress_bars.any? %>
+ <% if secondary_progress_bars.any? %>
- <% milestoneable.secondary_progress_bars.each do |progress_bar| %>
+ <% secondary_progress_bars.each do |progress_bar| %>
<%= progress_bar.title %>
diff --git a/app/components/milestones/progress_bars_component.rb b/app/components/milestones/progress_bars_component.rb
index 399005b3f..986385cec 100644
--- a/app/components/milestones/progress_bars_component.rb
+++ b/app/components/milestones/progress_bars_component.rb
@@ -6,11 +6,19 @@ class Milestones::ProgressBarsComponent < ApplicationComponent
end
def render?
- milestoneable.primary_progress_bar
+ primary_progress_bar
end
private
+ def primary_progress_bar
+ milestoneable.primary_progress_bar
+ end
+
+ def secondary_progress_bars
+ milestoneable.secondary_progress_bars
+ end
+
def progress_tag_for(progress_bar)
text = number_to_percentage(progress_bar.percentage, precision: 0)