class ActiveRecord::DisableJoinsAssociationRelation

def load
  super
  records = @records
  records_by_id = records.group_by do |record|
    record[key]
  end
  records = ids.flat_map { |id| records_by_id[id] }
  records.compact!
  @records = records
end