Can you push the shaft back in?
There is a long shaft that goes through the whole tranmission and can pop out when changing axles. The other end will go into the drivers side axle while the part sticking out should be inside the diff in the transmission itself.