Picked up a new Ofna buggy that came with the Castle Mamba Monster Motor.

I new to brushless cars but this thing is insane. Either way my buddy and i were ripping around the parking lot and herd a noise and my buddy said he saw a few sparks.

upon inspection we saw the bearing on the rear diff has come out at the back of the drive shaft. Ideas on what i should do/buy to upgrade. Im not even sure what buggy this exactly is.

Please chime in with any advice.