Set register button to correct color