Axolotl buckets infinite water bug(misunderstood earlier so posting again)(MCPE-123585)

steps to reproduce:
make a world on minecraft bedrock edition 1.17.0
get an axolotl(/summon, egg, bucket, naturally...)
find water(bucket, naturally)
when you take the axolotl with a bucket it doesnt remove the water and since bucket of axolotl is also a source it doubles the water making it infinite
possible fixes
make the axolotl only takeable when on a full water block
make the water block which axolotl was previously on disappear