1 files changed, 2 insertions, 2 deletions
diff --git a/app/models/work_items/widgets/hierarchy.rb b/app/models/work_items/widgets/hierarchy.rb
index dadd341de83..930aced8ace 100644
--- a/app/models/work_items/widgets/hierarchy.rb
+++ b/app/models/work_items/widgets/hierarchy.rb
@@ -4,13 +4,13 @@ module WorkItems
   module Widgets
     class Hierarchy < Base
       def parent
-        return unless Feature.enabled?(:work_items_hierarchy, work_item.project)
+        return unless work_item.project.work_items_feature_flag_enabled?
 
         work_item.work_item_parent
       end
 
       def children
-        return WorkItem.none unless Feature.enabled?(:work_items_hierarchy, work_item.project)
+        return WorkItem.none unless work_item.project.work_items_feature_flag_enabled?
 
         work_item.work_item_children
       end