In vanilla 1.17+, all buckets with axolotls look like they have the pink axolotl in them regardless of which type is actually inside. This client-side Fabric mod fixes this so that the item texture shows you what's actually inside it. It also adds textures for the baby forms, so you know exactly what's in each bucket.
The new textures look like this:
This mod is compatible with More Axolotl Variants and has textures for all the variants added by it.
REQUIRES FABRIC API