Push :3

enjarai · Sep 12, 2024 · e33319c · e33319c
1 parent 05bbfab
commit e33319c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/main/java/dev/enjarai/blahajtotem/BlahajFlags.java b/src/main/java/dev/enjarai/blahajtotem/BlahajFlags.java
@@ -1,14 +1,15 @@
 package dev.enjarai.blahajtotem;
 
 import dev.enjarai.blahajtotem.pond.BakedHuggableModel;
+import net.fabricmc.fabric.api.renderer.v1.model.WrapperBakedModel;
 import net.minecraft.client.MinecraftClient;
 import net.minecraft.entity.LivingEntity;
 import net.minecraft.item.ItemStack;
-import net.minecraft.item.Items;
 
 public class BlahajFlags {
     public static boolean isHuggable(ItemStack itemStack, LivingEntity entity) {
         var model = MinecraftClient.getInstance().getItemRenderer().getModel(itemStack, entity.getWorld(), entity, 0);
+        model = WrapperBakedModel.unwrap(model);
         return model instanceof BakedHuggableModel huggableModel && huggableModel.blahaj_totem$isHuggable();
     }
 }