diff --git a/lib/generators/noticed/model_generator.rb b/lib/generators/noticed/model_generator.rb
index 76ccff92..5ccdbc82 100644
--- a/lib/generators/noticed/model_generator.rb
+++ b/lib/generators/noticed/model_generator.rb
@@ -15,7 +15,7 @@ class ModelGenerator < Rails::Generators::NamedBase
       argument :attributes, type: :array, default: [], banner: "field:type field:type"
 
       def generate_notification
-        generate :model, name, "recipient:references{polymorphic}", "type", "params:json", "read_at:datetime:index", *attributes
+        generate :model, name, "recipient:references{polymorphic}", "type", "record:references{polymorphic}", "params:json", "read_at:datetime:index", *attributes
       end
 
       def add_noticed_model