Zombie and Husk stop converting to Drowned if removed from water after 30 seconds

Description:

Zombie and Husk stop converting to Drowned if removed from water after 30 seconds

Steps to reproduce:

1. Load Creative world

2. Make 2 block deep hole protected from sun

3. Fill it with water

4. Put Zombie or Husk in water

5. Wait 30 seconds until they start to shake

6. Remove water before 15 seconds of shaking has passed

Expected result:

Shaking should continue even if Zombie or Husk are removed from water after 30 seconds and after 15 second of shaking they should convert to Drowned

Actual result:

Shaking stops and Conversion to Drowned does not happen