vertex_shader: Fix control reaches end of function warning

But it could be changed in the future, and then you have this silent and annoying bug. Anyhow, maybe a switch instead then?