diff --git a/PCK-Studio/Controls/PckEditor.cs b/PCK-Studio/Controls/PckEditor.cs
index 54487926..d6a0ee10 100644
--- a/PCK-Studio/Controls/PckEditor.cs
+++ b/PCK-Studio/Controls/PckEditor.cs
@@ -935,6 +935,13 @@ namespace PckStudio.Controls
                         break;
                     }
 
+                    if (resourceLocation.Category == ResourceCategory.MobEntityTextures || resourceLocation.Category == ResourceCategory.ItemEntityTextures)
+                    {
+                        buttonEdit.Text = "EDIT ENTITY MODEL";
+                        buttonEdit.Visible = true;
+                        break;
+                    }
+
                     buttonEdit.Text = "EDIT TEXTURE ATLAS";
                     buttonEdit.Visible = true;
                 }