Fix category selection