1 files changed, 15 insertions, 2 deletions
diff --git a/src/main/java/de/hysky/skyblocker/mixins/GenericContainerScreenHandlerMixin.java b/src/main/java/de/hysky/skyblocker/mixins/GenericContainerScreenHandlerMixin.java
index f75af09a..bd066fc9 100644
--- a/src/main/java/de/hysky/skyblocker/mixins/GenericContainerScreenHandlerMixin.java
+++ b/src/main/java/de/hysky/skyblocker/mixins/GenericContainerScreenHandlerMixin.java
@@ -2,7 +2,10 @@ package de.hysky.skyblocker.mixins;
 
 import de.hysky.skyblocker.SkyblockerMod;
 import de.hysky.skyblocker.skyblock.dungeon.partyfinder.PartyFinderScreen;
+import de.hysky.skyblocker.skyblock.item.SkyblockInventoryScreen;
 import net.minecraft.client.MinecraftClient;
+import net.minecraft.client.gui.screen.Screen;
+import net.minecraft.client.gui.screen.ingame.GenericContainerScreen;
 import net.minecraft.item.ItemStack;
 import net.minecraft.screen.GenericContainerScreenHandler;
 import net.minecraft.screen.ScreenHandler;
@@ -22,8 +25,18 @@ public abstract class GenericContainerScreenHandlerMixin extends ScreenHandler {
     public void setStackInSlot(int slot, int revision, ItemStack stack) {
         super.setStackInSlot(slot, revision, stack);
         SkyblockerMod.getInstance().containerSolverManager.markDirty();
-        if (MinecraftClient.getInstance().currentScreen instanceof PartyFinderScreen screen) {
-            screen.markDirty();
+
+        Screen currentScreen = MinecraftClient.getInstance().currentScreen;
+        switch (currentScreen) {
+            case PartyFinderScreen screen -> screen.markDirty();
+            case GenericContainerScreen screen when screen.getTitle().getString().toLowerCase().contains("equipment") -> {
+                int line = slot/9;
+                if (line > 0 && line < 5 && slot % 9 == 1) {
+                    boolean empty = stack.getName().getString().trim().toLowerCase().startsWith("empty");
+                    SkyblockInventoryScreen.equipment[line - 1] = empty ? ItemStack.EMPTY : stack;
+                }
+            }
+            case null, default -> {}
         }
     }