The present teachings provide a method for controlling transmission of power to a set of wheels of a vehicle. The method can include providing a drive module configured to provide an amount of drive torque for powering the set of vehicle wheels. The method can include determining a yaw rate of the vehicle and a first set of vehicle parameters. The method can include determining a reference yaw rate of the vehicle based on the first set of vehicle parameters. The method can include calculating a yaw rate error based on the yaw rate and the reference yaw rate. The method can include reducing the amount of drive torque provided by the drive module to the vehicle wheels based on the yaw rate error.