Right now I have a custom trim material which works fine without the mod, but with the mod it makes the trim color on the item model use the diamond palette instead of my own (the armor model works as normal of course). Is there something I need to provide in the resource pack or data pack so the color shows properly?