diff --git a/main/src/main/java/net/citizensnpcs/trait/AttributeTrait.java b/main/src/main/java/net/citizensnpcs/trait/AttributeTrait.java
index f0babec23..590f25cb4 100644
--- a/main/src/main/java/net/citizensnpcs/trait/AttributeTrait.java
+++ b/main/src/main/java/net/citizensnpcs/trait/AttributeTrait.java
@@ -14,7 +14,7 @@ import net.citizensnpcs.api.trait.TraitName;
 
 @TraitName("attributetrait")
 public class AttributeTrait extends Trait {
-    @Persist
+    @Persist(keyType = Attribute.class)
     private final Map<Attribute, Double> attributes = Maps.newEnumMap(Attribute.class);
 
     public AttributeTrait() {