Use proper grain fragment shader