A device that includes a memory and a processor is disclosed. The processor may be configured to receive a control signal for operating a plurality of traction motors of a work machine. The control signal may include information relating to an actual speed of the work machine, a target speed of the work machine, and a generator speed of a generator operatively coupled to the traction motors. The processor may be configured to determine respective torque commands associated with the traction motors based on the actual speed and the target speed, and determine a generator power limit based on the generator speed. The processor may be configured to determine a threshold based on the respective torque commands and the generator power limit, adjust the respective torque commands based on the threshold, and cause the traction motors to be operated based on the adjusted respective torque commands.