Fixed drive button behaviour

Previously needed drive button to be pressed twice after a fully completed rest period.