Fix in shader code