@Override
  public int getArmorDisplay(EntityPlayer player, ItemStack armor, int slot) {
    NBTTagCompound nbt = getOrCreateTagCompound(armor);
    for (int i = 0; i < nbt.getInteger("moduleCount"); i++) {
      IArmorModule module = (IArmorModule) ModuleRegistry.getModule(nbt.getString("module" + i));

      if ((module != null) && (module.getArmorEffectType() == EnumArmorEffectType.DEFENSIVE)) {
        IDefensiveArmorModule defmodule = (IDefensiveArmorModule) module;
        return defmodule.getArmorToDisplay();
      }
    }
    return 0;
  }