From b92345295d1df3c79d7b21a43ecf4ac1f2df0fd7 Mon Sep 17 00:00:00 2001
From: Des Herriott <des.herriott@gmail.com>
Date: Thu, 18 Apr 2024 09:33:14 +0100
Subject: [PATCH] fix reach boost staying active after chestplate removed

#1301
---
 .../pneumatic_armor/handlers/ReachDistanceHandler.java | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/src/main/java/me/desht/pneumaticcraft/common/pneumatic_armor/handlers/ReachDistanceHandler.java b/src/main/java/me/desht/pneumaticcraft/common/pneumatic_armor/handlers/ReachDistanceHandler.java
index 4836d784f..bef501878 100644
--- a/src/main/java/me/desht/pneumaticcraft/common/pneumatic_armor/handlers/ReachDistanceHandler.java
+++ b/src/main/java/me/desht/pneumaticcraft/common/pneumatic_armor/handlers/ReachDistanceHandler.java
@@ -75,8 +75,16 @@ public void onToggle(ICommonArmorHandler commonArmorHandler, boolean newState) {
         if (!newState) {
             AttributeInstance attr = commonArmorHandler.getPlayer().getAttribute(ForgeMod.BLOCK_REACH.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);
+        }
+    }
 }