Terrain Trees Replacement Shader

Hello everyone!

I am trying to implement different cameras simulation (thermal, night vision etc).
To each object we set a variable with temperature and a replacement shader to the camera. It works fine with Objects I put manually, BUT it doen't work with PAINTED trees (terrain trees).

Do you have a suggestion how to make the shader working with painted trees? Thanks in advance!