HLSL shader fix