How do I make custom fish buckets for variants?

I'm making a mod for a snail mob and I want the snail variants to be save but when I use NBTs for items it doesnt carry over when I try to transfer that data values to the actual entity when you place it down