```ruby
require "tag"

module TagRepository

  # If tag_type parameter is given, load all tags of that type.
  def self.load_all(options={})
    tag_type = options[:tag_type]
    if tag_type.nil?
      Tag.all
    else
      Tag.where(tag_type: tag_type)
    end
  end

  def self.load_all_with_ids(ids)
    Tag.where(tag_id: ids)
  end

  def self.load(id)
    Tag.where(tag_id: id).first
  end

  def self.load_all_top_level(options = {})
    options[:parent_id] = nil
    Tag.where(options)
  end

  def self.put(tag)
    t = Tag.where(tag_id: tag[:tag_id]).first
    unless t
      Tag.create! tag
    else
      t.update_attributes! tag
    end
  end
end
```

Use correct query method when loading all tags by id
user
```ruby
require "tag"

module TagRepository

  # If tag_type parameter is given, load all tags of that type.
  def self.load_all(options={})
    tag_type = options[:tag_type]
    if tag_type.nil?
      Tag.all
    else
      Tag.where(tag_type: tag_type)
    end
  end

  def self.load_all_with_ids(ids)
    Tag.any_in(tag_id: ids)
  end

  def self.load(id)
    Tag.where(tag_id: id).first
  end

  def self.load_all_top_level(options = {})
    options[:parent_id] = nil
    Tag.where(options)
  end

  def self.put(tag)
    t = Tag.where(tag_id: tag[:tag_id]).first
    unless t
      Tag.create! tag
    else
      t.update_attributes! tag
    end
  end
end
```
assistant