Add AI Based on capability, not type

AtomicBlom · Sep 2, 2016 · de3bd41 · de3bd41
1 parent 030bbd6
commit de3bd41
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/build.properties b/build.properties
@@ -1,5 +1,5 @@
 # suppress inspection "UnusedProperty" for whole file
-mod_version=1.1.1
+mod_version=1.1.2
 mod_id=shearmadness
 group_name=com.github.atomicblom.shearmadness
 

diff --git a/src/main/java/com/github/atomicblom/shearmadness/proxy/CommonForgeEventProxy.java b/src/main/java/com/github/atomicblom/shearmadness/proxy/CommonForgeEventProxy.java
@@ -171,7 +171,7 @@ public void onCapabilityAttaching(AttachCapabilitiesEvent.Entity event)
     public void onCommonEntityJoinWorldEvent(EntityJoinWorldEvent event)
     {
         final Entity entity = event.getEntity();
-        if (entity instanceof EntitySheep)
+        if (entity.hasCapability(CapabilityProvider.CHISELED_SHEEP, null))
         {
             final EntityLiving livingEntity = (EntityLiving) event.getEntity();
             final EntityAITasks tasks = livingEntity.tasks;