Fix villager restock time display only works for librarians (#133)

Signed-off-by: Hendrix-Shen <[email protected]>
plusls · Oct 26, 2024 · 0927682 · 0927682
1 parent a8f4798
commit 0927682
Showing 1 changed file with 2 additions and 1 deletion.
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
@@ -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;
         }