Fix #605

Corosauce · Dec 3, 2023 · 4e01c67 · 4e01c67
1 parent 0b171cd
commit 4e01c67
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/main/java/weather2/blockentity/DeflectorBlockEntity.java b/src/main/java/weather2/blockentity/DeflectorBlockEntity.java
@@ -8,6 +8,7 @@
 import net.minecraft.world.level.block.state.BlockState;
 import weather2.ServerTickHandler;
 import weather2.WeatherBlocks;
+import weather2.weathersystem.WeatherManagerServer;
 
 public class DeflectorBlockEntity extends BlockEntity {
 
@@ -38,6 +39,7 @@ public void init() {
     @Override
     public void setRemoved() {
         super.setRemoved();
-        ServerTickHandler.getWeatherManagerFor(level).removeDeflector(getBlockPos());
+        WeatherManagerServer weatherManagerServer = ServerTickHandler.getWeatherManagerFor(level);
+        if (weatherManagerServer != null) weatherManagerServer.removeDeflector(getBlockPos());
     }
 }