Axolotl when put in a bucket loses it's colour

How to reproduce:
1. Spawn an axolotl
2. Put it in bucket (in survival)
3. Put it back down

What I expected to happen: axolotl retains it's colour

What happened: colour is randomised