Small correction in z axis scaling