[1.16.4]How to add custom enchantment with custom enchantmentType to creative tab like normal enchantment?

[HeidFanatic]

Hello. I've registered an enchantment which can be applied to sword and axe:

public class CustomEnchantment extends Enchantment{
    public CustomEnchantment(Rarity p_i46731_1_) {
        super(p_i46731_1_,EnchantmentType.create("AXE_AND_SWORD", (item)->(item instanceof SwordItem || item instanceof AxeItem)), new EquipmentSlotType[]{EquipmentSlotType.MAINHAND});
	}
}

But it doesn't show up in either Tool Tab or BattleGear Tab like other enchantment. The only way to find it in creative mode is to search it via searchbar.

How can I fix it?

Thank you so much!

Edited April 24, 2021 by HeidFanatic
Add Polite Word

[[NoOneButNo]]

Get all the enchantment category from

ItemGroup#getEnchantmentCategories

Then add yours (preferably you want to convert the array into a list...)

After that, set the enchantment category to the one you have just updated. 

ItemGroup#setEnchantmentCategories