extend torque control with mass balance