Also bear in mind the thermostat housing is different, and as such the hoses are different too. You will need to change either the housing or the hoses and rad pipe.
An alteration needs to be made to the coolant temp switch as it's in a different place on the 4x4 head (extend the wiring).
And the oil feed to the turbo comes out the exhaust side of the 4x4 head so either blank it off and use the block feed or you need the 4x4 oil feed pipe.
And the oil filter housing location is different as 4x4 have the remote arm that 2wd don't have.
Small differences but they will cause an issue if not thought about before putting it in place.
Hope this helps.
cheeRS