Use correct classes, to get the correct option chosen