From f80ed2d1e8afc1d32dcd83c0ffb50203b0ae1f72 Mon Sep 17 00:00:00 2001
From: paulevsGitch <paulevs@yandex.ru>
Date: Mon, 19 Oct 2020 01:47:51 +0300
Subject: [PATCH] Move functions back

---
 .../java/ru/betterend/mixin/common/DimensionTypeMixin.java   | 5 -----
 .../ru/betterend/world/generator/BetterEndBiomeSource.java   | 4 ++++
 2 files changed, 4 insertions(+), 5 deletions(-)

diff --git a/src/main/java/ru/betterend/mixin/common/DimensionTypeMixin.java b/src/main/java/ru/betterend/mixin/common/DimensionTypeMixin.java
index 7c560ec6..e8d69efe 100644
--- a/src/main/java/ru/betterend/mixin/common/DimensionTypeMixin.java
+++ b/src/main/java/ru/betterend/mixin/common/DimensionTypeMixin.java
@@ -11,8 +11,6 @@ import net.minecraft.world.dimension.DimensionType;
 import net.minecraft.world.gen.chunk.ChunkGenerator;
 import net.minecraft.world.gen.chunk.ChunkGeneratorSettings;
 import net.minecraft.world.gen.chunk.NoiseChunkGenerator;
-import ru.betterend.registry.BlockTagRegistry;
-import ru.betterend.util.FeaturesHelper;
 import ru.betterend.world.generator.BetterEndBiomeSource;
 
 @Mixin(value = DimensionType.class, priority = 100)
@@ -20,9 +18,6 @@ public class DimensionTypeMixin
 {
 	@Inject(method = "createEndGenerator", at = @At("HEAD"), cancellable = true)
 	private static void replaceGenerator(Registry<Biome> biomeRegistry, Registry<ChunkGeneratorSettings> chunkGeneratorSettingsRegistry, long seed, CallbackInfoReturnable<ChunkGenerator> info) {
-		BlockTagRegistry.addTerrainTags(biomeRegistry);
-		FeaturesHelper.addFeatures(biomeRegistry);
-		
 		info.setReturnValue(new NoiseChunkGenerator(new BetterEndBiomeSource(biomeRegistry, seed), seed, () -> {
 			return (ChunkGeneratorSettings) chunkGeneratorSettingsRegistry.getOrThrow(ChunkGeneratorSettings.END);
 		}));
diff --git a/src/main/java/ru/betterend/world/generator/BetterEndBiomeSource.java b/src/main/java/ru/betterend/world/generator/BetterEndBiomeSource.java
index b47505c9..31794461 100644
--- a/src/main/java/ru/betterend/world/generator/BetterEndBiomeSource.java
+++ b/src/main/java/ru/betterend/world/generator/BetterEndBiomeSource.java
@@ -17,6 +17,8 @@ import net.minecraft.world.biome.source.TheEndBiomeSource;
 import net.minecraft.world.gen.ChunkRandom;
 import ru.betterend.BetterEnd;
 import ru.betterend.registry.BiomeRegistry;
+import ru.betterend.registry.BlockTagRegistry;
+import ru.betterend.util.FeaturesHelper;
 import ru.betterend.world.biome.EndBiome;
 
 public class BetterEndBiomeSource extends BiomeSource {
@@ -48,6 +50,8 @@ public class BetterEndBiomeSource extends BiomeSource {
 		this.noise = new SimplexNoiseSampler(chunkRandom);
 
 		BiomeRegistry.mutateRegistry(biomeRegistry);
+		BlockTagRegistry.addTerrainTags(biomeRegistry);
+		FeaturesHelper.addFeatures(biomeRegistry);
 	}
 	
 	private static List<Biome> getBiomes(Registry<Biome> biomeRegistry) {