Fix computeScreenSpacePosition for 3D