After replacing the transmission it will not shift into 4-wheel drive?

I would make sure the electrical connector to the shift actuator is secure and clean with no damaged pins. Here is the location of the actuator on the transfer case so you can see what I am talking about. Check out the images (below). Please let us know what happens.