Add back tool mode to wrench (#2462)

1 files changed, 12 insertions, 0 deletions

diff --git a/src/main/java/gregtech/common/tools/GT_Tool_Wrench.java b/src/main/java/gregtech/common/tools/GT_Tool_Wrench.java
index 91ecd71536..0b992a2710 100644
--- a/src/main/java/gregtech/common/tools/GT_Tool_Wrench.java
+++ b/src/main/java/gregtech/common/tools/GT_Tool_Wrench.java
@@ -30,6 +30,7 @@ import gregtech.api.enums.Textures;
 import gregtech.api.interfaces.IIconContainer;
 import gregtech.api.items.GT_MetaGenerated_Tool;
 import gregtech.api.util.GT_ToolHarvestHelper;
+import gregtech.common.items.behaviors.Behaviour_Switch_Mode;
 import gregtech.common.items.behaviors.Behaviour_Wrench;
 import ic2.api.tile.IWrenchable;
 
@@ -149,6 +150,7 @@ public class GT_Tool_Wrench extends GT_Tool {
 
     @Override
     public void onStatsAddedToTool(GT_MetaGenerated_Tool aItem, int aID) {
+        aItem.addItemBehavior(aID, new Behaviour_Switch_Mode());
         aItem.addItemBehavior(aID, new Behaviour_Wrench(100));
     }
 
@@ -256,4 +258,14 @@ public class GT_Tool_Wrench extends GT_Tool {
         }
         return modified;
     }
+
+    @Override
+    public byte getMaxMode() {
+        return 2;
+    }
+
+    @Override
+    public String getToolTypeName() {
+        return "wrench";
+    }
 }