From 1360b4171c4272756c663afcdda81ed5dccb522e Mon Sep 17 00:00:00 2001
From: paulevsGitch <paulevs@yandex.ru>
Date: Mon, 2 Nov 2020 18:11:36 +0300
Subject: [PATCH] More fixes

---
 gradle.properties                                            | 2 +-
 src/main/java/ru/betterend/registry/EndStructures.java       | 2 +-
 .../java/ru/betterend/world/structures/piece/LakePiece.java  | 5 +++++
 3 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/gradle.properties b/gradle.properties
index c5d8d362..52370fe9 100644
--- a/gradle.properties
+++ b/gradle.properties
@@ -8,7 +8,7 @@
 	loader_version = 0.10.5+build.213
 
 # Mod Properties
-	mod_version = 0.5.1-beta
+	mod_version = 0.5.2-beta
 	maven_group = ru.betterend
 	archives_base_name = better-end
 
diff --git a/src/main/java/ru/betterend/registry/EndStructures.java b/src/main/java/ru/betterend/registry/EndStructures.java
index ca7cd908..c96039e0 100644
--- a/src/main/java/ru/betterend/registry/EndStructures.java
+++ b/src/main/java/ru/betterend/registry/EndStructures.java
@@ -44,7 +44,7 @@ public class EndStructures {
 	}
 	
 	public static void registerBiomeStructures(Identifier id, Biome biome, Collection<Supplier<ConfiguredStructureFeature<?, ?>>> structures) {
-		if (!id.getPath().contains("mountain")) {
+		if (!id.getPath().contains("mountain") && !id.getPath().contains("lake")) {
 			addStructure(ETERNAL_PORTAL, structures);
 		}
 	}
diff --git a/src/main/java/ru/betterend/world/structures/piece/LakePiece.java b/src/main/java/ru/betterend/world/structures/piece/LakePiece.java
index 34d6e91a..bd1b9001 100644
--- a/src/main/java/ru/betterend/world/structures/piece/LakePiece.java
+++ b/src/main/java/ru/betterend/world/structures/piece/LakePiece.java
@@ -120,12 +120,17 @@ public class LakePiece extends BasePiece {
 								}
 							}
 							maxY = MHelper.randRange(2, 3, random);
+							int last = maxY - 1;
 							for (int i = 0; i < maxY; i++) {
 								pos.setY(pos.getY() - 1);
 								BlockState state = chunk.getBlockState(pos);
 								if (state.getMaterial().isReplaceable() || state.isIn(EndTags.GEN_TERRAIN)) {
 									if (pos.getY() > 56) {
 										chunk.setBlockState(pos, AIR, false);
+										if (pos.getY() == last) {
+											state = world.getBiome(pos.add(sx, 0, sz)).getGenerationSettings().getSurfaceConfig().getTopMaterial();
+											chunk.setBlockState(pos.down(), state, false);
+										}
 									}
 									else if (pos.getY() == 56) {
 										if (random.nextBoolean()) {