Fix a couple issues with entity categories

froobynooby · Apr 28, 2024 · 5833a84 · 5833a84
1 parent 700e936
commit 5833a84
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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());
     }
 
 }