diff --git a/src/main/java/me/gallowsdove/foxymachines/abstracts/AbstractWand.java b/src/main/java/me/gallowsdove/foxymachines/abstracts/AbstractWand.java
index b1a1bff..84c7e5f 100644
--- a/src/main/java/me/gallowsdove/foxymachines/abstracts/AbstractWand.java
+++ b/src/main/java/me/gallowsdove/foxymachines/abstracts/AbstractWand.java
@@ -81,7 +81,8 @@ public void preRegister() {
     protected ItemUseHandler onUse() {
         return e -> {
             Player player = e.getPlayer();
-            ItemStack itemInInventory = player.getInventory().getItemInMainHand();
+            ItemStack itemInInventory = e.getItem();
+            if (itemInInventory == null) return;
             ItemMeta meta = itemInInventory.getItemMeta();
             PersistentDataContainer container = meta.getPersistentDataContainer();