Updated enchantment code