Fix shader
Fix color gradient