diff --git a/src/main/java/net/coreprotect/utility/entity/EntityUtil.java b/src/main/java/net/coreprotect/utility/entity/EntityUtil.java
index dcf43b8..98f8fe7 100644
--- a/src/main/java/net/coreprotect/utility/entity/EntityUtil.java
+++ b/src/main/java/net/coreprotect/utility/entity/EntityUtil.java
@@ -338,7 +338,7 @@ public class EntityUtil {
                             }
                         }
                         else if (count == 1) {
-                            if (abstractVillager instanceof Villager && value instanceof Villager.Type) {
+                            if (abstractVillager instanceof Villager && (value instanceof Villager.Type || value instanceof String)) {
                                 Villager villager = (Villager) abstractVillager;
                                 if (value instanceof String) {
                                     value = BukkitAdapter.ADAPTER.getRegistryValue((String) value, Villager.Type.class);