diff --git a/lib/noticed/model.rb b/lib/noticed/model.rb
index ba70dd35..44c57c0a 100644
--- a/lib/noticed/model.rb
+++ b/lib/noticed/model.rb
@@ -20,6 +20,7 @@ module Model
       end
 
       belongs_to :recipient, polymorphic: true
+      belongs_to :record, polymorphic: true
 
       scope :newest_first, -> { order(created_at: :desc) }
       scope :unread, -> { where(read_at: nil) }