Small fix in the vertex shader