lib/approval/models/approval/comment.rb in approval-0.3.1 vs lib/approval/models/approval/comment.rb in approval-0.3.2

@@ -1,11 +1,11 @@
 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
+      def define_user_association
+        belongs_to :user, class_name: Approval.config.user_class_name
       end
     end
 
     belongs_to :request, class_name: :"Approval::Request", inverse_of: :comments