diff --git a/main/src/main/java/net/citizensnpcs/trait/shop/NPCShopAction.java b/main/src/main/java/net/citizensnpcs/trait/shop/NPCShopAction.java index 39f9356f3..9dc94c0eb 100644 --- a/main/src/main/java/net/citizensnpcs/trait/shop/NPCShopAction.java +++ b/main/src/main/java/net/citizensnpcs/trait/shop/NPCShopAction.java @@ -1,13 +1,14 @@ package net.citizensnpcs.trait.shop; -import java.util.Map; -import java.util.WeakHashMap; +import java.util.List; import java.util.function.Consumer; import java.util.function.Supplier; import org.bukkit.entity.Entity; import org.bukkit.inventory.ItemStack; +import com.google.common.collect.Lists; + import net.citizensnpcs.api.gui.InventoryMenuPage; import net.citizensnpcs.api.persistence.PersistenceLoader; import net.citizensnpcs.api.persistence.PersisterRegistry; @@ -79,15 +80,15 @@ public abstract class NPCShopAction implements Cloneable { } public static Iterable getGUIs() { - return GUI.values(); + return GUI; } public static void register(Class clazz, String type, GUI gui) { REGISTRY.register(type, clazz); - GUI.put(clazz, gui); + GUI.add(gui); } - private static final Map, GUI> GUI = new WeakHashMap<>(); + private static final List GUI = Lists.newArrayList(); private static final PersisterRegistry REGISTRY = PersistenceLoader .createRegistry(NPCShopAction.class); } \ No newline at end of file