This TT01 metal drive shaft stablizer can help prevent metal drive shaft wobbling. Simply use super glue to attach it to the chassis.

Download: Upgradeparts-tam-tt01-shaft-stablizer v1