There are many shafts that go through until the power of the motor is transmitted to the tire, but these bearings are made of special resin plastic bearings.
Replacing this with a ball bearing is the first step in improving performance and you can expect less resistance to increase speed and riding time.
Part for TA06 drive system
(From Amazon.co.jp)