This isn't really an onroad engine question but it should be pretty easy

The screw in the diff must be all most all the way tight to keep pressure on the large rings and balls in the main gear or else it will spin.

Be sure you didn't lose any of the balls or thrust assembly, look at the pages in this manual from teamlosi.com

XXX-NT Manual