A drive system for a power machine, comprising: a drive pump; a drive motor operably coupled to and driven by the drive pump, the drive motor having a rotational output member; a gearbox having an input shaft operably coupled to the rotational output member of the drive motor and an output shaft operably coupled to a wheel, the gearbox further including: a plurality of gear reduction sets (232, 234) coupled to the output shaft; and a coupling mechanism for coupling the input shaft to the output shaft via engagement with a selected one of the plurality of gear reduction sets (232, 234); a gearshift input manipulable by an operator to provide an indication of a selected gear reduction set; a sensor configured to provide an engagement feedback signal indicative of whether the coupling mechanism is engaged with the selected gear reduction set; a controller in communication with the gearshift input to receive the indication of the selected gear reduction set and the sensor to receive the engagement feedback signal, and characterized in that when the controller determines, based on the received engagement feedback signal, that the coupling mechanism is not engaged with the selected gear reduction set, the controller is configured to provide an alignment signal to the drive pump to control the rotational output member and align the coupling mechanism with the selected gear reduction set from the plurality of gear reduction sets (232, 234), and wherein the controller is configured to stop providing the alignment signal to the drive pump when the sensor indicates that the coupling mechanism is engaged with the selected gear reduction set, to prevent inadvertent movement of the power machine. a drive pump; a drive motor operably coupled to and driven by the drive pump, the drive motor having a rotational output member; a gearbox having an input shaft operably coupled to the rotational output member of the drive motor and an output shaft operably coupled to a wheel, the gearbox further including: a plurality of gear reduction sets (232, 234) coupled to the output shaft; and a coupling mechanism for coupling the input shaft to the output shaft via engagement with a selected one of the plurality of gear reduction sets (232, 234); a plurality of gear reduction sets (232, 234) coupled to the output shaft; and a coupling mechanism for coupling the input shaft to the output shaft via engagement with a selected one of the plurality of gear reduction sets (232, 234); a gearshift input manipulable by an operator to provide an indication of a selected gear reduction set; a sensor configured to provide an engagement feedback signal indicative of whether the coupling mechanism is engaged with the selected gear reduction set; a controller in communication with the gearshift input to receive the indication of the selected gear reduction set and the sensor to receive the engagement feedback signal, and characterized in that when the controller determines, based on the received engagement feedback signal, that the coupling mechanism is not engaged with the selected gear reduction set, the controller is configured to provide an alignment signal to the drive pump to control the rotational output member and align the coupling mechanism with the selected gear reduction set from the plurality of gear reduction sets (232, 234), and wherein the controller is configured to stop providing the alignment signal to the drive pump when the sensor indicates that the coupling mechanism is engaged with the selected gear reduction set, to prevent inadvertent movement of the power machine.