Baby axolotl's die of suffocation when swimming against a bottom of a block

The baby axolotl's die of suffocation when they swim upwards into the bottom of a block, you can recreate it by getting the water source to connect to the block above it, then if a baby axolotl swims into it they suffocate since their hitbox clips into the block