Small update to category selection