diff --git a/main/src/main/java/net/citizensnpcs/npc/CitizensTraitFactory.java b/main/src/main/java/net/citizensnpcs/npc/CitizensTraitFactory.java index 497b45d0e..07155da12 100644 --- a/main/src/main/java/net/citizensnpcs/npc/CitizensTraitFactory.java +++ b/main/src/main/java/net/citizensnpcs/npc/CitizensTraitFactory.java @@ -1,5 +1,6 @@ package net.citizensnpcs.npc; +import java.util.Collection; import java.util.List; import java.util.Map; import java.util.Set; @@ -132,6 +133,11 @@ public class CitizensTraitFactory implements TraitFactory { registered.remove(info.getTraitName()); } + @Override + public Collection getRegisteredTraits() { + return registered.values(); + } + @Override public T getTrait(Class clazz) { for (TraitInfo entry : registered.values()) {