Constructor fix in DriveTrain