diff --git a/src/main/java/com/gregtechceu/gtceu/common/machine/electric/HullMachine.java b/src/main/java/com/gregtechceu/gtceu/common/machine/electric/HullMachine.java index 1cd99cd3cfe..96100c9eac2 100644 --- a/src/main/java/com/gregtechceu/gtceu/common/machine/electric/HullMachine.java +++ b/src/main/java/com/gregtechceu/gtceu/common/machine/electric/HullMachine.java @@ -38,7 +38,7 @@ public class HullMachine extends TieredPartMachine implements IMonitorComponent public HullMachine(BlockEntityCreationInfo info, int tier) { super(info, tier); if (GTCEu.Mods.isAE2Loaded()) { - this.gridNodeHost = attachTrait(new GridNodeHostTrait(this)); + this.gridNodeHost = GridNodeHostTransformer.attachToMachine(this); } else { this.gridNodeHost = null; } @@ -82,6 +82,10 @@ public void setFrontFacing(Direction facing) { private static class GridNodeHostTransformer implements ValueTransformer { + private static Object attachToMachine(HullMachine machine) { + return machine.attachTrait(new GridNodeHostTrait(machine)); + } + @Override public Tag serializeNBT(Object value, TransformerContext context) { if (GTCEu.Mods.isAE2Loaded() &&