diff --git a/src/main/java/ru/betterend/blocks/BlockEndLotusSeed.java b/src/main/java/ru/betterend/blocks/BlockEndLotusSeed.java index 3db2190e..383b52f3 100644 --- a/src/main/java/ru/betterend/blocks/BlockEndLotusSeed.java +++ b/src/main/java/ru/betterend/blocks/BlockEndLotusSeed.java @@ -63,6 +63,12 @@ public class BlockEndLotusSeed extends BlockUnderwaterPlantWithAge { BlocksHelper.setWithoutUpdate(world, bpos, flower); bpos.setY(bpos.getY() - 1); stem = world.getBlockState(bpos); + if (stem.isOf(BlockRegistry.END_LOTUS_STEM)) { + stem = BlockRegistry.END_LOTUS_STEM.getDefaultState(); + if (!world.getBlockState(bpos.north()).getFluidState().isEmpty()) { + stem = stem.with(BlockEndLotusStem.WATERLOGGED, true); + } + } BlocksHelper.setWithoutUpdate(world, bpos, stem.with(BlockEndLotusStem.SHAPE, TripleShape.TOP)); } }