Update Page.java

Author: coderFlameyosFlow

src/main/java/me/flame/menus/menu/Page.java

@@ -5,6 +5,7 @@
 import me.flame.menus.items.MenuItem;
 import me.flame.menus.menu.iterator.PageIterator;
 
+import org.bukkit.inventory.Inventory;
 import org.bukkit.inventory.ItemStack;
 import org.jetbrains.annotations.Contract;
 import org.jetbrains.annotations.NotNull;
@@ -153,9 +154,16 @@ public void removeItem(MenuItem... items) {
         Set<MenuItem> slots = ImmutableSet.copyOf(items);
 
         int size = itemMap.size();
+        Inventory inventory = holder.getInventory();
         for (int i = 0; i < size; i++) {
             MenuItem item = itemMap.get(i);
-            if (item != null && slots.contains(item)) itemMap.remove(i);
+            if (item == null) continue;
+
+            ItemStack itemStack = item.getItemStack();
+            if (slots.contains(item)) {
+                itemMap.remove(i);
+                inventory.remove(itemStack);
+            }
         }
     }
 
@@ -167,9 +175,16 @@ public void removeItem(ItemStack... items) {
         Set<ItemStack> slots = ImmutableSet.copyOf(items);
 
         int size = itemMap.size();
+        Inventory inventory = holder.getInventory();
         for (int i = 0; i < size; i++) {
             MenuItem item = itemMap.get(i);
-            if (item != null && slots.contains(item.getItemStack())) itemMap.remove(i);
+            if (item == null) continue;
+
+            ItemStack itemStack = item.getItemStack();
+            if (slots.contains(itemStack)) {
+                itemMap.remove(i);
+                inventory.remove(itemStack);
+            }
         }
     }
 
@@ -366,4 +381,4 @@ public boolean hasItem(Slot slot) {
         if (!slot.isSlot()) return false;
         return itemMap.get(slot.slot) != null;
     }
-}
+}