Fixed issues in the DriveTrain