Fixed unity missing for source z dimension. Tests have shown the results are the same.