From c0322423db4d9e977cc7c109dfdd73bb2316f6b5 Mon Sep 17 00:00:00 2001
From: paulevsGitch <paulevs@yandex.ru>
Date: Sun, 13 Dec 2020 16:44:22 +0300
Subject: [PATCH] Registry fixes

---
 src/main/java/ru/betterend/registry/EndBiomes.java | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/main/java/ru/betterend/registry/EndBiomes.java b/src/main/java/ru/betterend/registry/EndBiomes.java
index 586e2b05..97660a05 100644
--- a/src/main/java/ru/betterend/registry/EndBiomes.java
+++ b/src/main/java/ru/betterend/registry/EndBiomes.java
@@ -163,6 +163,11 @@ public class EndBiomes {
 			RegistryKey<Biome> key = picker.pickRandom(random);
 			FABRIC_VOID.add(key.getValue());
 		}
+		picker = InternalBiomeData.getEndBiomesMap().get(BiomeKeys.END_BARRENS);
+		for (int i = 0; i < 1000; i++) {
+			RegistryKey<Biome> key = picker.pickRandom(random);
+			FABRIC_VOID.add(key.getValue());
+		}
 		if (BetterEnd.isDevEnvironment()) {
 			System.out.println("Added void biomes from Fabric API:");
 			FABRIC_VOID.forEach((id) -> {
@@ -329,6 +334,7 @@ public class EndBiomes {
 		float weight = biome.getGenChanceImmutable();
 		RegistryKey<Biome> key = BuiltinRegistries.BIOME.getKey(biome.getBiome()).get();
 		InternalBiomeData.addEndBiomeReplacement(BiomeKeys.END_HIGHLANDS, key, weight);
+		InternalBiomeData.addEndBiomeReplacement(BiomeKeys.END_MIDLANDS, key, weight);
 	}
 	
 	private static void addVoidBiomeToFabricApi(EndBiome biome) {