diff --git a/common/src/main/java/net/blay09/mods/balm/api/BalmHooks.java b/common/src/main/java/net/blay09/mods/balm/api/BalmHooks.java
index 289e95ea..92d391e5 100644
--- a/common/src/main/java/net/blay09/mods/balm/api/BalmHooks.java
+++ b/common/src/main/java/net/blay09/mods/balm/api/BalmHooks.java
@@ -12,6 +12,7 @@
 import net.minecraft.world.entity.Pose;
 import net.minecraft.world.entity.player.Player;
 import net.minecraft.world.item.DyeColor;
+import net.minecraft.world.item.Item;
 import net.minecraft.world.item.ItemStack;
 import net.minecraft.world.level.Level;
 import net.minecraft.world.level.block.state.BlockState;
@@ -56,7 +57,7 @@ default CompoundTag getPersistentData(Player player) {
 
     int getBurnTime(ItemStack itemStack);
 
-    void setBurnTime(ItemStack itemStack, int burnTime);
+    void setBurnTime(Item item, int burnTime);
 
     void firePlayerCraftingEvent(Player player, ItemStack crafted, Container craftMatrix);
 
diff --git a/fabric/src/main/java/net/blay09/mods/balm/fabric/FabricBalmHooks.java b/fabric/src/main/java/net/blay09/mods/balm/fabric/FabricBalmHooks.java
index 173d59cc..83b564f4 100644
--- a/fabric/src/main/java/net/blay09/mods/balm/fabric/FabricBalmHooks.java
+++ b/fabric/src/main/java/net/blay09/mods/balm/fabric/FabricBalmHooks.java
@@ -105,8 +105,8 @@ public int getBurnTime(ItemStack itemStack) {
     }
 
     @Override
-    public void setBurnTime(ItemStack itemStack, int burnTime) {
-        FuelRegistry.INSTANCE.add(itemStack.getItem(), burnTime);
+    public void setBurnTime(Item item, int burnTime) {
+        FuelRegistry.INSTANCE.add(item, burnTime);
     }
 
     @Override
diff --git a/forge/src/main/java/net/blay09/mods/balm/forge/ForgeBalmHooks.java b/forge/src/main/java/net/blay09/mods/balm/forge/ForgeBalmHooks.java
index 3961217a..743777d9 100644
--- a/forge/src/main/java/net/blay09/mods/balm/forge/ForgeBalmHooks.java
+++ b/forge/src/main/java/net/blay09/mods/balm/forge/ForgeBalmHooks.java
@@ -119,8 +119,8 @@ public int getBurnTime(ItemStack itemStack) {
     }
 
     @Override
-    public void setBurnTime(ItemStack itemStack, int burnTime) {
-        burnTimes.put(itemStack.getItem(), burnTime);
+    public void setBurnTime(Item item, int burnTime) {
+        burnTimes.put(item, burnTime);
     }
 
     @Override
diff --git a/neoforge/src/main/java/net/blay09/mods/balm/neoforge/NeoForgeBalmHooks.java b/neoforge/src/main/java/net/blay09/mods/balm/neoforge/NeoForgeBalmHooks.java
index e20f8ffa..dc5af2ff 100644
--- a/neoforge/src/main/java/net/blay09/mods/balm/neoforge/NeoForgeBalmHooks.java
+++ b/neoforge/src/main/java/net/blay09/mods/balm/neoforge/NeoForgeBalmHooks.java
@@ -117,8 +117,8 @@ public int getBurnTime(ItemStack itemStack) {
     }
 
     @Override
-    public void setBurnTime(ItemStack itemStack, int burnTime) {
-        burnTimes.put(itemStack.getItem(), burnTime);
+    public void setBurnTime(Item item, int burnTime) {
+        burnTimes.put(item, burnTime);
     }
 
     @Override