Axolotl's in buckets when placed do not remember variant

When you pick up an Axolotl in a bucket and place it it will not remember the variant it was before