lib/pupper/auditable.rb in pupper-0.1.0 vs lib/pupper/auditable.rb in pupper-0.1.1

@@ -29,10 +29,10 @@
         run_callbacks :update, &block
         changes_applied
       end
 
       def audit_logs
-        AuditLog.where(auditable_type: model_name.name, auditable_id: primary_key)
+        audit_model.where(auditable_type: model_name.name, auditable_id: primary_key)
       end
 
       def create_audit_log
         return unless changed?