Merge pull request #7 from Aranyalma2/master

Fix crashing by turtle unequip
PolymerPorts · Jun 2, 2024 · d88119d · d88119d
2 parents 5ed76d0 + bfc0f6d
commit d88119d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/main/java/eu/pb4/cctpatch/impl/poly/model/TurtleModel.java b/src/main/java/eu/pb4/cctpatch/impl/poly/model/TurtleModel.java
@@ -124,6 +124,9 @@ public void setUpgrades(TurtleBrain brain, ITurtleUpgrade left, ITurtleUpgrade r
     }
 
     private ItemStack getUpgradeModel(ITurtleUpgrade upgrade, TurtleBrain brain, TurtleSide turtleSide) {
+        if(upgrade == null){
+            return ItemStack.EMPTY;
+        }
         if (upgrade instanceof TurtleModemAccessor modem) {
             var type = modem.isAdvanced() ? ADVANCED_MODEM_MODEL : NORMAL_MODEM_MODEL;
             var x = brain.getUpgradeData(turtleSide).get(ModRegistry.DataComponents.ON.get());