Some baby mobs having swimming issues

Baby chickens will not swim up when under water and drown. Baby pigs do swim up but do not get high enough out of the water and drown. I haven't tested all baby mobs, but I have tested baby sheep, baby pigs, baby horses, baby cows and baby chickens.