updating for drivetrain_design type bug