diff --git a/electrical/src/main/java/resonantinduction/electrical/tesla/TeslaGrid.java b/electrical/src/main/java/resonantinduction/electrical/tesla/TeslaGrid.java index ef3d45d2e..1fb9efbb6 100644 --- a/electrical/src/main/java/resonantinduction/electrical/tesla/TeslaGrid.java +++ b/electrical/src/main/java/resonantinduction/electrical/tesla/TeslaGrid.java @@ -3,9 +3,10 @@ */ package resonantinduction.electrical.tesla; -import java.util.HashSet; +import java.util.Collections; import java.util.Iterator; import java.util.Set; +import java.util.WeakHashMap; import mffs.api.fortron.IServerThread; import net.minecraft.server.ServerListenThread; @@ -21,7 +22,7 @@ public class TeslaGrid private static final TeslaGrid INSTANCE_CLIENT = new TeslaGrid(); private static final TeslaGrid INSTANCE_SERVER = new TeslaGrid(); - private final Set tileEntities = new HashSet(); + private final Set tileEntities = Collections.newSetFromMap(new WeakHashMap()); public void register(ITesla iTesla) { diff --git a/electrical/src/main/java/resonantinduction/electrical/tesla/TileTesla.java b/electrical/src/main/java/resonantinduction/electrical/tesla/TileTesla.java index 864141f0f..3a87c9d80 100644 --- a/electrical/src/main/java/resonantinduction/electrical/tesla/TileTesla.java +++ b/electrical/src/main/java/resonantinduction/electrical/tesla/TileTesla.java @@ -32,6 +32,7 @@ import calclavia.lib.multiblock.reference.IMultiBlockStructure; import calclavia.lib.multiblock.reference.MultiBlockHandler; import calclavia.lib.network.IPacketReceiver; import calclavia.lib.network.IPacketSender; +import calclavia.lib.network.PacketHandler; import calclavia.lib.prefab.tile.TileElectrical; import calclavia.lib.render.EnumColor; @@ -187,7 +188,7 @@ public class TileTesla extends TileElectrical implements IMultiBlockStructure