fix count method signature for active_record60

- refs 95ffb25 - 95ffb25
flyerhzm · Jan 17, 2024 · f063cad · f063cad
1 parent f15e7a1
commit f063cad
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lib/bullet/active_record60.rb b/lib/bullet/active_record60.rb
@@ -282,7 +282,7 @@ def count_records
 
       ::ActiveRecord::Associations::CollectionProxy.prepend(
         Module.new do
-          def count
+          def count(column_name = nil)
             if Bullet.start? && !proxy_association.is_a?(::ActiveRecord::Associations::ThroughAssociation)
               Bullet::Detector::CounterCache.add_counter_cache(
                 proxy_association.owner,
@@ -293,7 +293,7 @@ def count
                 proxy_association.reflection.name
               )
             end
-            super
+            super(column_name)
           end
         end
       )