module Bullet::ActiveRecord

def find(*args)
  result = origin_find(*args)
  if Bullet.start?
    if result.is_a? Array
      Bullet::Detector::NPlusOneQuery.add_possible_objects(result)
      Bullet::Detector::CounterCache.add_possible_objects(result)
    elsif result.is_a? ::ActiveRecord::Base
      Bullet::Detector::NPlusOneQuery.add_impossible_object(result)
      Bullet::Detector::CounterCache.add_impossible_object(result)
    end
  end
  result
end