diff --git a/app/services/affinity_groups_hierarchy.rb b/app/services/affinity_groups_hierarchy.rb
index 5ac8e5b..bd0a467 100644
--- a/app/services/affinity_groups_hierarchy.rb
+++ b/app/services/affinity_groups_hierarchy.rb
@@ -17,7 +17,7 @@ class AffinityGroupsHierarchy < Array
       hydrate(group)
     end
 
-    load_discomforts
+    discomforts
     freeze
   end
 
@@ -58,9 +58,9 @@ class AffinityGroupsHierarchy < Array
 
   private
 
-  def load_discomforts
-    @load_discomforts ||= GroupAffinity.pluck(:group_a_id, :group_b_id,
-                                              :discomfort).each_with_object({}) do |(id_a, id_b, discomfort), acc|
+  def discomforts
+    @discomforts ||= GroupAffinity.pluck(:group_a_id, :group_b_id,
+                                         :discomfort).each_with_object({}) do |(id_a, id_b, discomfort), acc|
       acc[uuid_to_int(id_a) + uuid_to_int(id_b)] = discomfort
     end
   end