app/models/ahoy/message.rb in ahoy_email-0.3.2 vs app/models/ahoy/message.rb in ahoy_email-0.4.0

@@ -1,7 +1,7 @@
 module Ahoy
   class Message < ActiveRecord::Base
     self.table_name = "ahoy_messages"
 
-    belongs_to :user, polymorphic: true
+    belongs_to :user, AhoyEmail.belongs_to.merge(polymorphic: true)
   end
 end