Entities no longer showing flame animation when in water/lava at the same time

The bug

In 1.16.5, when an armor stand is in water and lava at the same time, it shows a burning animation, while in the latest snapshot 21w05a the behavior is changed so that there's no flaming animation.

The first two photos with flame show the behavior in 1.16.5, and the last 3 without flame show the same setup in 21w05a.