CIT Enchantment applying on armor and elytra

#15

1 files changed, 3 insertions, 1 deletions

diff --git a/src/main/java/shcm/shsupercm/fabric/citresewn/pack/cits/CITEnchantment.java b/src/main/java/shcm/shsupercm/fabric/citresewn/pack/cits/CITEnchantment.java
index cce2f29..2458646 100644
--- a/src/main/java/shcm/shsupercm/fabric/citresewn/pack/cits/CITEnchantment.java
+++ b/src/main/java/shcm/shsupercm/fabric/citresewn/pack/cits/CITEnchantment.java
@@ -157,7 +157,7 @@ public class CITEnchantment extends CIT {
                     layer.build(false));
         }
 
-        public VertexConsumer tryApply(VertexConsumer base, VertexConsumerProvider provider) {
+        public VertexConsumer tryApply(VertexConsumer base, RenderLayer baseLayer, VertexConsumerProvider provider) {
             if (!shouldApply || appliedContext == null)
                 return null;
 
@@ -165,6 +165,8 @@ public class CITEnchantment extends CIT {
                     .map(cit -> provider.getBuffer(cit.renderLayers.get(GlintRenderLayer.this)))
                     .toArray(VertexConsumer[]::new));
 
+            provider.getBuffer(baseLayer); // refresh base layer for armor consumer
+
             return base == null ? applied : VertexConsumers.union(applied, base);
         }
     }