diff --git a/src/main/java/dev/micle/totemofreviving/item/crafting/ExtendedShapedRecipeBuilder.java b/src/main/java/dev/micle/totemofreviving/item/crafting/ExtendedShapedRecipeBuilder.java
index 9bfea9b..10dac03 100644
--- a/src/main/java/dev/micle/totemofreviving/item/crafting/ExtendedShapedRecipeBuilder.java
+++ b/src/main/java/dev/micle/totemofreviving/item/crafting/ExtendedShapedRecipeBuilder.java
@@ -12,6 +12,7 @@ import net.minecraft.advancements.critereon.RecipeUnlockedTrigger;
 import net.minecraft.data.recipes.FinishedRecipe;
 import net.minecraft.resources.ResourceLocation;
 import net.minecraft.tags.Tag;
+import net.minecraft.tags.TagKey;
 import net.minecraft.world.item.Item;
 import net.minecraft.world.item.crafting.Ingredient;
 import net.minecraft.world.item.crafting.RecipeSerializer;
@@ -45,8 +46,8 @@ public class ExtendedShapedRecipeBuilder {
 		return new ExtendedShapedRecipeBuilder(serializer, result, count);
 	}
 	
-	public ExtendedShapedRecipeBuilder define(Character symbol, Tag<Item> tagIn) {
-		return this.define(symbol, Ingredient.of((ItemLike) tagIn));
+	public ExtendedShapedRecipeBuilder define(Character symbol, TagKey<Item> tagIn) {
+		return this.define(symbol, Ingredient.of(tagIn));
 	}
 	
 	public ExtendedShapedRecipeBuilder define(Character symbol, ItemLike itemIn) {