module Bullet::Mongoid

def set_relation(name, relation)
  if relation && relation.metadata.macro !~ /embed/
    Bullet::Detector::NPlusOneQuery.call_association(self, name)
  end
  origin_set_relation(name, relation)
end