Fix logic in DriveTrain update.