This probably does it: https://github.com/jk3064/Custom-Unit-Shader-Framework/pull/1 (also see Flove)