lib/approval/models/approval/comment.rb in approval-0.2.3 vs lib/approval/models/approval/comment.rb in approval-0.3.0

@@ -1,10 +1,15 @@
 module Approval
   class Comment < ::ActiveRecord::Base
     self.table_name_prefix = "approval_".freeze
 
+    class << self
+      def define_user_association(klass)
+        belongs_to :user, class_name: klass.to_s
+      end
+    end
+
     belongs_to :request, class_name: :"Approval::Request", inverse_of: :comments
-    belongs_to :user,    class_name: Approval.config.user_class_name
 
     validates :content, presence: true, length: { maximum: Approval.config.comment_maximum }
   end
 end