diff --git a/src/main/java/xyz/alexcrea/cuanvil/enchant/wrapped/CABukkitEnchantment.java b/src/main/java/xyz/alexcrea/cuanvil/enchant/wrapped/CABukkitEnchantment.java
index 024e04e..6b87d7d 100644
--- a/src/main/java/xyz/alexcrea/cuanvil/enchant/wrapped/CABukkitEnchantment.java
+++ b/src/main/java/xyz/alexcrea/cuanvil/enchant/wrapped/CABukkitEnchantment.java
@@ -134,8 +134,7 @@ public class CABukkitEnchantment extends CAEnchantmentBase {
     }
 
     private static EnchantmentRarity findRarity(Enchantment enchantment) {
-        //TODO use non deprecated value
-        return EnchantmentRarity.getRarity(enchantment.getRarity().getWeight());
+        return EnchantmentRarity.getRarity(enchantment.getAnvilCost());
     }
 
     @Override