Pixel shader fix