fixing drivetrain bug