diff --git a/src/main/java/com/froobworld/farmcontrol/utils/EntityCategory.java b/src/main/java/com/froobworld/farmcontrol/utils/EntityCategory.java
index 28d08f8..6f21732 100644
--- a/src/main/java/com/froobworld/farmcontrol/utils/EntityCategory.java
+++ b/src/main/java/com/froobworld/farmcontrol/utils/EntityCategory.java
@@ -63,7 +63,7 @@ public boolean isMember(Class<?> entityClass) {
     }
 
     public boolean isMember(SnapshotEntity entity) {
-        return isMember(entity.getEntityClass());
+        return isMember(entity.getEntityType());
     }
 
     public boolean isMember(EntityType entityType) {
@@ -90,7 +90,7 @@ public boolean isCompatibleWith(Action action) {
     }
 
     public static EntityCategory ofName(String categoryName) {
-        return EntityCategory.entityCategoryMap.get(categoryName);
+        return EntityCategory.entityCategoryMap.get(categoryName.toLowerCase());
     }
 
 }