A computer program product stored on a non-transitory computer-readable medium, said computer program product for controlling operation of a vehicle, wherein the method is performed by a control arrangement, and wherein the vehicle comprises two or more powertrains each comprising a power source configured to provide motive power to the vehicle via wheels arranged at a respective wheel axle of the vehicle, wherein each of the two or more powertrains is separate from, and independent of, the other of the two or more powertrains, and wherein said computer program product comprising computer instructions to cause a control arrangement to perform the following operations:
- performing a gear shift in a transmission of one of the two or more powertrains;
- and
- controlling the power source of at least one other powertrain of the two or more powertrains to compensate for at least part of an interruption of a wheel torque obtained during the step of performing the gear shift.