From 00f04eeb4e4657d394e79bd6b1bb56f80467fae5 Mon Sep 17 00:00:00 2001
From: Andrew Brown <drusepth@gmail.com>
Date: Wed, 12 Feb 2020 01:30:13 -0600
Subject: [PATCH] include promo users in premium badges

---
 app/views/thredded/users/_badge.html.erb | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/app/views/thredded/users/_badge.html.erb b/app/views/thredded/users/_badge.html.erb
index 6124e8c5..7a9722a8 100644
--- a/app/views/thredded/users/_badge.html.erb
+++ b/app/views/thredded/users/_badge.html.erb
@@ -10,7 +10,7 @@
     <span style="<%= badge_style %>; background: #dedeff"><%= badge_text.presence || 'Beta Tester' %></span>
   <% elsif user.selected_billing_plan_id == 3 %>
     <span style="<%= badge_style %>; background: #ffdede"><%= badge_text.presence || 'Early Adopter' %></span>
-  <% elsif [4, 5, 6].include? user.selected_billing_plan_id %>
+  <% elsif user.on_premium_plan? %>
     <span style="<%= badge_style %>; background: #ffffaa"><%= badge_text.presence || 'Premium Supporter' %></span>
   <% end %>
 <% end %>