how to add selected cell manually to stored (added)

[MehdiAloui]

I create a react component for a scheduling grid . It allows users to select specific hours across days in a week, with features for selecting/deselecting entire rows (days) and columns (hours). The problem is when i select from the extra row or column it will select the specified column or row but after i try deselect one cell from this row or column it wont only after the second click on it.
this my onmove code

const onMove = ({ selection, store: { changed: { added, removed } } }) => {
    setSelected((prev) => {
      const next = new Set(prev);
      extractIds(added).forEach((id) => next.add(id));
      extractIds(removed).forEach((id) => next.delete(id));
      return next.size !== prev.size ||
        [...next].some((id) => !prev.has(id)) ||
        [...prev].some((id) => !next.has(id))
        ? next
        : prev;
    });
  };

and this 2 functions to handle the behaviour

 const handleColumnSelect = (colIndex) => {
    const newColumns = new Set(extraSelected.columns);
    const newSelection = new Set(selected);

    if (newColumns.has(colIndex)) {
      newColumns.delete(colIndex);
      for (let row = 0; row < 7; row++) newSelection.delete(row * 24 + colIndex);
    } else {
      newColumns.add(colIndex);
      for (let row = 0; row < 7; row++) newSelection.add(row * 24 + colIndex);
    }

    setExtraSelected((prev) => ({ ...prev, columns: newColumns }));
    setSelected(newSelection);
  };

const handleRowSelect = (rowIndex) => {
   const newRows = new Set(extraSelected.rows);
   const newSelection = new Set(selected);

   if (newRows.has(rowIndex)) {
     newRows.delete(rowIndex);
     for (let col = 0; col < 24; col++) newSelection.delete(rowIndex * 24 + col);
   } else {
     newRows.add(rowIndex);
     for (let col = 0; col < 24; col++) newSelection.add(rowIndex * 24 + col);
   }

   setExtraSelected((prev) => ({ ...prev, rows: newRows }));
   setSelected(newSelection);
 };

[simonwep]

Hey! Could you share a minimal, full reproduction? It's pretty difficult to find out exactly what went wrong without a whole example :)