diff --git a/src/main/java/org/betterx/betterend/blocks/EndLotusFlowerBlock.java b/src/main/java/org/betterx/betterend/blocks/EndLotusFlowerBlock.java
index 6fa832b8..723487d8 100644
--- a/src/main/java/org/betterx/betterend/blocks/EndLotusFlowerBlock.java
+++ b/src/main/java/org/betterx/betterend/blocks/EndLotusFlowerBlock.java
@@ -11,6 +11,7 @@ import net.minecraft.world.item.ItemStack;
 import net.minecraft.world.level.BlockGetter;
 import net.minecraft.world.level.block.Block;
 import net.minecraft.world.level.block.state.BlockState;
+import net.minecraft.world.level.material.MapColor;
 import net.minecraft.world.level.storage.loot.LootParams;
 import net.minecraft.world.phys.shapes.CollisionContext;
 import net.minecraft.world.phys.shapes.VoxelShape;
@@ -27,7 +28,7 @@ public class EndLotusFlowerBlock extends EndPlantBlock implements BehaviourPlant
     private static final VoxelShape SHAPE_COLLISION = Block.box(0, 0, 0, 16, 2, 16);
 
     public EndLotusFlowerBlock() {
-        super(BehaviourBuilders.createPlant().lightLevel((bs) -> 15));
+        super(BehaviourBuilders.createPlant(MapColor.COLOR_PINK, true).lightLevel((bs) -> 15));
     }
 
     @Override
diff --git a/src/main/java/org/betterx/betterend/blocks/EndLotusLeafBlock.java b/src/main/java/org/betterx/betterend/blocks/EndLotusLeafBlock.java
index 4362526d..accad1ce 100644
--- a/src/main/java/org/betterx/betterend/blocks/EndLotusLeafBlock.java
+++ b/src/main/java/org/betterx/betterend/blocks/EndLotusLeafBlock.java
@@ -23,6 +23,7 @@ import net.minecraft.world.level.block.state.BlockState;
 import net.minecraft.world.level.block.state.StateDefinition;
 import net.minecraft.world.level.block.state.properties.BlockStateProperties;
 import net.minecraft.world.level.block.state.properties.EnumProperty;
+import net.minecraft.world.level.material.MapColor;
 import net.minecraft.world.level.material.WaterFluid;
 import net.minecraft.world.phys.shapes.CollisionContext;
 import net.minecraft.world.phys.shapes.VoxelShape;
@@ -36,7 +37,7 @@ public class EndLotusLeafBlock extends BaseBlockNotFull implements RenderLayerPr
     private static final VoxelShape VSHAPE = Block.box(0, 0, 0, 16, 1, 16);
 
     public EndLotusLeafBlock() {
-        super(BehaviourBuilders.createPlant().noOcclusion().sound(SoundType.WET_GRASS));
+        super(BehaviourBuilders.createPlant(MapColor.PLANT, true).noOcclusion().sound(SoundType.WET_GRASS));
     }
 
     @Override
diff --git a/src/main/java/org/betterx/betterend/world/structures/piece/LakePiece.java b/src/main/java/org/betterx/betterend/world/structures/piece/LakePiece.java
index b84dc10f..f36921ba 100644
--- a/src/main/java/org/betterx/betterend/world/structures/piece/LakePiece.java
+++ b/src/main/java/org/betterx/betterend/world/structures/piece/LakePiece.java
@@ -140,8 +140,11 @@ public class LakePiece extends BasePiece {
                                 world,
                                 worldPos
                         )) {
-                            state = chunk.getBlockState(mut.above());
-                            if (state.isAir()) {
+                            state = chunk.getBlockState(mut.above(3));
+                            final BlockState stateAbove = chunk.getBlockState(mut.above());
+                            if (stateAbove.isAir() && state.isAir()) {
+                                state = random.nextInt(10) == 0 ? ENDSTONE : EndBiome.findTopMaterial(world, worldPos);
+                            } else if (stateAbove.isAir()) {
                                 state = random.nextBoolean() ? ENDSTONE : EndBiome.findTopMaterial(world, worldPos);
                             } else {
                                 state = state.getFluidState().isEmpty()