From 7c810cd3a7f24249b22ae346ed5395a5b57c8366 Mon Sep 17 00:00:00 2001
From: Frank <developer@ambertation.de>
Date: Tue, 14 Jun 2022 11:58:41 +0200
Subject: [PATCH] [Fix] Crashes with FABI 0.56 (#4)

---
 .../betterx/bclib/mixin/common/TheEndBiomeDataMixin.java    | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/main/java/org/betterx/bclib/mixin/common/TheEndBiomeDataMixin.java b/src/main/java/org/betterx/bclib/mixin/common/TheEndBiomeDataMixin.java
index 8455e400..1215fc36 100644
--- a/src/main/java/org/betterx/bclib/mixin/common/TheEndBiomeDataMixin.java
+++ b/src/main/java/org/betterx/bclib/mixin/common/TheEndBiomeDataMixin.java
@@ -32,14 +32,14 @@ public class TheEndBiomeDataMixin implements TheEndBiomeDataAccessor {
     private Map<Holder<Biome>, WeightedPicker<Holder<Biome>>> endBarrensMap;
 
     public boolean bcl_canGenerateAsEndBiome(ResourceKey<Biome> key) {
-        return endBiomesMap != null && endBiomesMap.containsKey(key);
+        return endBiomesMap != null && endBiomesMap.keySet().stream().map(h->h.unwrapKey().orElse(null)).anyMatch(k->k!=null && k.equals(key));
     }
 
     public boolean bcl_canGenerateAsEndMidlandBiome(ResourceKey<Biome> key) {
-        return endMidlandsMap != null && endMidlandsMap.containsKey(key);
+        return endMidlandsMap != null && endMidlandsMap.keySet().stream().map(h->h.unwrapKey().orElse(null)).anyMatch(k->k!=null && k.equals(key));
     }
 
     public boolean bcl_canGenerateAsEndBarrensBiome(ResourceKey<Biome> key) {
-        return endBarrensMap != null && endBarrensMap.containsKey(key);
+        return endBarrensMap != null && endBarrensMap.keySet().stream().map(h->h.unwrapKey().orElse(null)).anyMatch(k->k!=null && k.equals(key));
     }
 }