1 files changed, 3 insertions, 3 deletions
diff --git a/app/models/event.rb b/app/models/event.rb
index b54fbe6f7e8..a617694b7dd 100644
--- a/app/models/event.rb
+++ b/app/models/event.rb
@@ -279,15 +279,15 @@ class Event < ActiveRecord::Base
   end
 
   def note_commit?
-    target.noteable_type == "Commit"
+    target.for_commit?
   end
 
   def issue_note?
-    note? && target && target.noteable_type == "Issue"
+    note? && target && target.for_issue?
   end
 
   def note_project_snippet?
-    target.noteable_type == "Snippet"
+    target.for_snippet?
   end
 
   def note_target