1 files changed, 7 insertions, 2 deletions
diff --git a/lib/api/helpers/discussions_helpers.rb b/lib/api/helpers/discussions_helpers.rb
index 799d5582b38..3c0db1d0ea9 100644
--- a/lib/api/helpers/discussions_helpers.rb
+++ b/lib/api/helpers/discussions_helpers.rb
@@ -3,10 +3,15 @@
 module API
   module Helpers
     module DiscussionsHelpers
-      def self.noteable_types
+      def self.feature_category_per_noteable_type
         # This is a method instead of a constant, allowing EE to more easily
         # extend it.
-        [Issue, Snippet, MergeRequest, Commit]
+        {
+          Issue => :issue_tracking,
+          Snippet => :snippets,
+          MergeRequest => :code_review,
+          Commit => :code_review
+        }
       end
     end
   end