v2.0.4

.gitignore

@@ -17,4 +17,5 @@ out/
 
 # Visual Studio Code
 .history/
-Icon*
+Icon*
+!icon.png

gradle.properties

@@ -2,7 +2,7 @@ org.gradle.jvmargs=-Xmx3G
 org.gradle.daemon=false
 mc_version=1.18.2
 
-mod_version = 2.0.2
+mod_version = 2.0.4
 
 create_version = 0.5.+
 flywheel_version = 0.6.+

src/main/java/com/forsteri/createendertransmission/blocks/energyTransmitter/EnergyTransmitterBlockEntity.java

@@ -42,19 +42,27 @@ public List<BlockPos> addPropagationLocations(IRotate block, BlockState state, L
     public List<KineticBlockEntity> getConnectedTransmitters(){
         Map<String, List<KineticBlockEntity>> channel = EnergyNetwork.ENERGY.channels.get(getChannel());
 
+        channel.values().removeIf(List::isEmpty);
+
+        channel.forEach((password, transmitters) -> transmitters.removeIf(blockEntity ->
+                !(blockEntity instanceof EnergyTransmitterBlockEntity transmitter)
+                        ||
+                                transmitter.getChannel() != getChannel() || !transmitter.getPassword().equals(password)
+                        ));
+
+        channel.values().forEach(list -> list.removeIf(BlockEntity::isRemoved));
+
         if (channel.containsKey(getPassword())) {
             List<KineticBlockEntity> list = channel
                     .get(getPassword());
 
             if(!list.contains(this))
                 list.add(this);
 
-            return list;
-        }
 
-        channel.values().removeIf(List::isEmpty);
 
-        channel.values().forEach(list -> list.removeIf(BlockEntity::isRemoved));
+            return list;
+        }
 
         ArrayList<KineticBlockEntity> list = new ArrayList<>(List.of(this));
 
@@ -72,17 +80,16 @@ public void reloadSettings(){
 
         if (level == null) return;
 
+        if (hasNetwork())
+            getOrCreateNetwork().remove(this);
         detachKinetics();
+        removeSource();
     }
 
     @Override
     public void afterReload(){
         if (level == null) return;
 
-        if (hasNetwork())
-            getOrCreateNetwork().remove(this);
-        removeSource();
-        detachKinetics();
         attachKinetics();
     }