Very simular to MC-213794.

On Bedrock Edition, Azalea Leaves or Flowering Azalea Leaves can be waterlogged. This behaviour does not exist on Java Edition.

A solution would be to add a waterlogged block state to Azalea Leaves and Flowering Azalea Leaves.