diff --git a/src/main/java/com/songoda/ultimatestacker/lootables/LootablesManager.java b/src/main/java/com/songoda/ultimatestacker/lootables/LootablesManager.java
index 234579d..c00e4f9 100644
--- a/src/main/java/com/songoda/ultimatestacker/lootables/LootablesManager.java
+++ b/src/main/java/com/songoda/ultimatestacker/lootables/LootablesManager.java
@@ -50,7 +50,7 @@ public class LootablesManager {
 
     private List<Drop> runLoot(LivingEntity entity, Loot loot, int rerollChance, int looting) {
         Modify modify = null;
-        if (entity.getType() == EntityType.SHEEP) {
+        if (entity instanceof Sheep) {
             modify = (Loot loot2) -> {
                 CompatibleMaterial material = loot2.getMaterial();
                 if (material.name().contains("WOOL") && ((Sheep) entity).getColor() != null) {