fix reach boost staying active after chestplate removed

#1301
TeamPneumatic · Apr 18, 2024 · 853f934 · 853f934
1 parent 2c6597a
commit 853f934
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/...in/java/me/desht/pneumaticcraft/common/pneumatic_armor/handlers/ReachDistanceHandler.java b/...in/java/me/desht/pneumaticcraft/common/pneumatic_armor/handlers/ReachDistanceHandler.java
@@ -78,8 +78,16 @@ public void onToggle(ICommonArmorHandler commonArmorHandler, boolean newState) {
         if (!newState) {
             AttributeInstance attr = commonArmorHandler.getPlayer().getAttribute(ForgeMod.REACH_DISTANCE.get());
             if (attr != null) {
-                attr.removeModifier(ReachDistanceHandler.REACH_DIST_BOOST);
+                attr.removeModifier(REACH_DIST_BOOST);
             }
         }
     }
+
+    @Override
+    public void onShutdown(ICommonArmorHandler commonArmorHandler) {
+        AttributeInstance attr = commonArmorHandler.getPlayer().getAttribute(ForgeMod.BLOCK_REACH.get());
+        if (attr != null) {
+            attr.removeModifier(REACH_DIST_BOOST);
+        }
+    }
 }