From d983613872be51eb475eabac603ff350f82647e3 Mon Sep 17 00:00:00 2001
From: paulevsGitch <paulevs@yandex.ru>
Date: Sat, 4 Dec 2021 15:30:36 +0300
Subject: [PATCH] Small changes

---
 src/main/java/ru/bclib/world/biomes/BCLBiome.java | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/src/main/java/ru/bclib/world/biomes/BCLBiome.java b/src/main/java/ru/bclib/world/biomes/BCLBiome.java
index bb59bc15..37bfc640 100644
--- a/src/main/java/ru/bclib/world/biomes/BCLBiome.java
+++ b/src/main/java/ru/bclib/world/biomes/BCLBiome.java
@@ -27,7 +27,7 @@ public class BCLBiome {
 	private final ResourceLocation biomeID;
 	private final Biome biome;
 	
-	private Runnable surfaceInit;
+	private Consumer<Biome> surfaceInit;
 	private BCLBiome biomeParent;
 	private Biome actualBiome;
 	private BCLBiome edge;
@@ -232,8 +232,8 @@ public class BCLBiome {
 			structures.forEach(s -> BiomeAPI.addBiomeStructure(BiomeAPI.getBiomeKey(actualBiome), s));
 		}
 		
-		if (this.surfaceInit != null){
-			surfaceInit.run();
+		if (this.surfaceInit != null) {
+			surfaceInit.accept(actualBiome);
 		}
 	}
 	
@@ -320,8 +320,8 @@ public class BCLBiome {
 	 * @param surface {@link SurfaceRules.RuleSource} rule.
 	 */
 	public void setSurface(RuleSource surface) {
-		this.surfaceInit = () -> {
-			ResourceKey key = BiomeAPI.getBiomeKey(biome);
+		this.surfaceInit = (actualBiome) -> {
+			ResourceKey key = BiomeAPI.getBiomeKey(actualBiome);
 			BiomeAPI.addSurfaceRule(biomeID, SurfaceRules.ifTrue(SurfaceRules.isBiome(key), surface));
 		};
 	}