1 files changed, 6 insertions, 4 deletions
diff --git a/lib/api/helpers/notes_helpers.rb b/lib/api/helpers/notes_helpers.rb
index f61bcfe963e..6798c4d284b 100644
--- a/lib/api/helpers/notes_helpers.rb
+++ b/lib/api/helpers/notes_helpers.rb
@@ -5,10 +5,12 @@ module API
     module NotesHelpers
       include ::RendersNotes
 
-      def self.noteable_types
-        # This is a method instead of a constant, allowing EE to more easily
-        # extend it.
-        [Issue, MergeRequest, Snippet]
+      def self.feature_category_per_noteable_type
+        {
+          Issue => :issue_tracking,
+          MergeRequest => :code_review,
+          Snippet => :snippets
+        }
       end
 
       def update_note(noteable, note_id)