Change spec to select by class