4 speed or torqueflite??

I would leave it in, but take it back to whoever did the work and then fix it to stop coming out of gear. The hydraulic leak would be easier and cheaper to fix than the swap.