Prevent trees from generating above water in Rainforest biomes

[leiseg]

Explain your suggestion here.

Currently trees in Wilder Wild's Rainforest biome can generate above water. This causes them to not attach to any ground and therefore float in the air. Perhaps there is a way to prevent this and instead generate trees only around and not above water?

Additional context

Examples of what I am referring to:

[AViewFromTheTop]

This is a vanilla bug, using the old lake feature.
The problem is that the lake searches for nearby areas to generate, which can be outside the current chunk (feature generation has access to a 3x3 area.)
Because of this, trees would have already generated in a surrounding chunk, therefore when the lake generates, it cuts the floor out from beneath them.
I’m not sure what to replace this with, as I do genuinely like the lake feature in this biome.
If you have any suggestions, please let me know.