Azalea trees can generate in more than two blocks of water

The bug

Azalea trees can generate more than two blocks submerged in water. According to this comment by [Mojang] slicedlime, the maximum amount of water that an azalea tree should generate in is two blocks.

How to reproduce

Create a Single Biome world with the Lush Caves biome:

Seed: 1238268594428547397
Coordinates: /execute in minecraft:overworld run tp @s 806.25 62.64 834.17 1445.84 11.57