From afe2468ee7835c1e21aa07c26535a46c4128fc61 Mon Sep 17 00:00:00 2001
From: Goodbird <goodbirdmail@yandex.ru>
Date: Thu, 25 Jan 2024 11:32:11 +0300
Subject: [PATCH] Fixed model selection gui not opening

---
 .../goodbird/npcgecko/mixin/impl/MixinGeckoAddonClient.java    | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/main/java/com/goodbird/npcgecko/mixin/impl/MixinGeckoAddonClient.java b/src/main/java/com/goodbird/npcgecko/mixin/impl/MixinGeckoAddonClient.java
index db67dc9..ec75f02 100644
--- a/src/main/java/com/goodbird/npcgecko/mixin/impl/MixinGeckoAddonClient.java
+++ b/src/main/java/com/goodbird/npcgecko/mixin/impl/MixinGeckoAddonClient.java
@@ -2,6 +2,7 @@ package com.goodbird.npcgecko.mixin.impl;
 
 import com.goodbird.npcgecko.client.gui.GuiStringSelection;
 import com.goodbird.npcgecko.client.gui.SubGuiModelAnimation;
+import com.goodbird.npcgecko.entity.EntityCustomModel;
 import com.goodbird.npcgecko.mixin.IDataDisplay;
 import kamkeel.addon.client.GeckoAddonClient;
 import net.minecraft.client.Minecraft;
@@ -39,7 +40,7 @@ public class MixinGeckoAddonClient {
         if(!supportEnabled)
             return;
 
-        if(EntityList.getEntityString(entity).equals("npcgecko.CustomModelEntity") || EntityList.getEntityString(entity).equals("Geckolib Model")){
+        if(entity instanceof EntityCustomModel){
             creationScreen.addButton(new GuiNpcButton(202, creationScreen.guiLeft-60, creationScreen.guiTop+40, 180, 20, ((IDataDisplay)creationScreen.npc.display).getCustomModelData().getModel()));
         }
     }