hi
could be drive shaft, but would have to be really bad to make it shake
even when they knock and click on steering lock, they never really shake or cause vibration
unless its came apart slightly when the spring and ball joint were done and the balls have dropped out of the cv/drive shaft, that would make it pretty sloppy and might cause shake/vibration
I'd still return it to the garage though
cheers
joe