Bed stays occupied after a lightning turns the villager sleeping in it into a witch

The bug

When lightning hits a villager that is sleeping in a bed, the villager will turn into a witch, and the bed will stay occupied forever.

To reproduce

1. Place a bed
2. Spawn a villager next to it
3. Set time to night
4. Run /summon minecraft:lightning_bolt ~ ~ ~ while on the bed
5. The villager turns into a witch and leaves the bed
6. Right-click onto the bed
   → The message "This bed is occupied" appears

Code analysis

Code analysis and potential fix can be found in this comment.