Underwater music interrupts current music and does not stop playing after exiting water

Underwater music is broken on bedrock. You can dip your head underwater for two seconds, or half a second for that matter and poof! The current music fades away, quickly replaced by underwater music. Honestly I dont really mind this since the same happens for the nether, however this contradicts what happens on java as there underwater music only starts playing a few minutes after the current track is finished. What I do mind is, after exiting the water, it continues on until the track is completely finished. This makes the music feel extremely out of place, especially with the water sounds still in the background.
How to reproduce:
1. Enter a world. Wait until some general music is playing. E.g Sweden
2. Dive underwater (head must be submerged) and exit the water. The glitch occurs