Fixed z axis bug