Thanks a bunch! (Don't know why I didn't think of this before.... XD)
https://imgur.com/a/cg7ti
For anyone looking for the answer in the future, this is what I did.
In your ItemBlock class I put:
@Override
public boolean isValidArmor(ItemStack stack, EntityEquipmentSlot armorType, Entity entity) {
if(armorType == EntityEquipmentSlot.HEAD) {
return true;
}else {
return false;
}
}
(Change EntityEquipmentSlot.HEAD to the respective armor piece or just return true if you want it to equip in all slots.