Correct state management for reset button

Make sure we set the initial state in the same way each time.