Make a copy of your material function and modify the function after the Texture Sample (pic1).
The function in a material (pic2).
Hope that helps.