So long story short does anyone know how to put the front powered truck back in?

Problem I'm having is the little swivel knuckle u-joint thing on the truck has a square hole that of course mates with the square shaft off the motor. But how do you line them up?

I'm thinking the decoder has to be unscrewed from the frame?

Any advice would be greatly appreciated.