diff --git a/src/main/java/io/github/apickledwalrus/skriptgui/gui/GUI.java b/src/main/java/io/github/apickledwalrus/skriptgui/gui/GUI.java index 92f6062..5d1dc07 100644 --- a/src/main/java/io/github/apickledwalrus/skriptgui/gui/GUI.java +++ b/src/main/java/io/github/apickledwalrus/skriptgui/gui/GUI.java @@ -250,11 +250,9 @@ public void setItem(Object slot, @Nullable ItemStack item, boolean stealable, @N rawShape = rawShape.replaceFirst("\\+", "" + ch2); ch = ch2; } - if (consumer != null) { - slots.put(ch, consumer); - } else { // Just in case as we may be updating a slot - slots.remove(ch); - } + + // Although we may be adding null consumers, it lets us track what slots have been set + slots.put(ch, consumer); if (stealable) { stealableSlots.add(ch);