diff --git a/src/main/java/com/plusls/MasaGadget/impl/feature/entityInfo/VillagerNextRestockTimeInfo.java b/src/main/java/com/plusls/MasaGadget/impl/feature/entityInfo/VillagerNextRestockTimeInfo.java index 46728f3..bca36be 100644 --- a/src/main/java/com/plusls/MasaGadget/impl/feature/entityInfo/VillagerNextRestockTimeInfo.java +++ b/src/main/java/com/plusls/MasaGadget/impl/feature/entityInfo/VillagerNextRestockTimeInfo.java @@ -16,8 +16,9 @@ public static Component getInfo(@NotNull Villager villager) { long nextRestockTime; long nextWorkTime; long timeOfDay = villager.getLevel().getDayTime() % 24000; + VillagerProfession profession = villager.getVillagerData().getProfession(); - if (villager.getVillagerData().getProfession() != VillagerProfession.LIBRARIAN) { + if (profession == VillagerProfession.NONE || profession == VillagerProfession.NITWIT) { return null; }