Fix #15 by always using the ClientLevel.

Mrbysco · Oct 12, 2024 · 3df95d2 · 3df95d2
1 parent 72263d2
commit 3df95d2
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/common/src/main/java/com/mrbysco/justenoughprofessions/VillagerCache.java b/common/src/main/java/com/mrbysco/justenoughprofessions/VillagerCache.java
@@ -2,11 +2,11 @@
 
 import com.mrbysco.justenoughprofessions.platform.Services;
 import net.minecraft.client.Minecraft;
+import net.minecraft.client.multiplayer.ClientLevel;
 import net.minecraft.nbt.CompoundTag;
 import net.minecraft.world.entity.EntityType;
 import net.minecraft.world.entity.npc.Villager;
 import net.minecraft.world.entity.npc.VillagerProfession;
-import net.minecraft.world.level.Level;
 import org.jetbrains.annotations.Nullable;
 
 import java.util.Objects;
@@ -20,8 +20,7 @@ public static Villager getVillagerEntity(VillagerProfession profession) {
 		if (cachedVillager == null) {
 			CompoundTag nbt = new CompoundTag();
 			nbt.putString("id", Objects.requireNonNull(Services.PLATFORM.getEntityKey(EntityType.VILLAGER)).toString());
-			Minecraft mc = Minecraft.getInstance();
-			Level level = mc.hasSingleplayerServer() && mc.getSingleplayerServer() != null ? mc.getSingleplayerServer().getAllLevels().iterator().next() : mc.level;
+			ClientLevel level = Minecraft.getInstance().level;
 			if (level != null) {
 				Villager villager = (Villager) EntityType.loadEntityRecursive(nbt, level, Function.identity());
 				if (villager != null) {