Game Ready Baking Trim Sheet

Hi everyone!

I have a model where I applied some trim sheets with DECALMachine and I would like to bake it so I have a texture for games.
image

So I have this smart material and I linked it to the trim node so it has the texture going on it.
The problem is that DECALMachine is unwrapping the part that is not assigned to the trim into a small area in the UVmap. So when I bake it, the texture not assigned to the trim sheet is stretched like crazy

If you guys (or girls) got any idea on how I can make it work it would be awesome!!

Thanks and happy modeling!

Here is the node setup: