Bring back the key selection button