Use empty bucket to pick up axolotl works wrongly

Summary:
Use empty bucket to pick up axolotl works wrongly in many ways.
Steps to Reproduce:
Use empty bucket to pick up axolotl
Observed Results:
When you pick up an axolotl using empty bucket in water source block, the water source block doesn't disappear. And you can even pick up axolotl using empty in flowing water.
Expected Results:
You can not pick up an axolotl in flowing water, and when you pick up axolotl in water source block, the block should disappear.