Geckolib Texture Entity Not Saving

So, I want to make an entity that when spawned has a random texture. I used the Geckolib Plugin to randomly change the entity texture on spawn. Unfortunately, while it does work the first time, when I save then reenter the world all the entities go back to one default texture. Is there anything I can do to make the random entity texture save with the world?