Changed it so its now using DifferentialPilot, this should ensure that the wheel motors start and stop at the same time