small update for DriveTrain