The solution will be very similar to this one here. Except instead of converting tangents to points, you just want to zero out all x and y components. This will require first a deconstruct point component and then a construct point where you only carry over the z values.